2 releases

0.1.1 May 14, 2023
0.1.0 May 12, 2023

#604 in Machine learning

MIT license

23KB
495 lines

AIMM

AI Model Manager.

Terminology

AIMM Civitai HuggingFace
Repository Model (checkpoint) Repo
Revision Model version Revsion / commit

Use it

# Scan an existing directory to build a manifest file
aimm scan

# Install packages from aimm.json
aimm install

# Install a package from a download link (to `.`)
aimm install https://civitai.com/api/download/models/46137

# Install a package from a download link (to a specific place)
aimm install https://civitai.com/api/download/models/46137 models/Stable-diffusion

To be implemented for MVP

Cli


# Smart install (install to appropriate place, e.g. models/Lora, download previews, etc.)
aimm si https://civitai.com/models/13213

# Or, in short:
aimm install --smart https://civitai.com/models/13213

# Initialize a new project
aimm init

# Install from Git
aimm install https://github.com/AUTOMATIC1111/stable-diffusion-webui

# Install a model from Civitai web page
aimm install https://civitai.com/models/7240/meinamix 

# Install a Lora from Civitai, and automatically pair it with previews
aimm install https://civitai.com/models/13213

# Install a model from Hugging Face
aimm install https://huggingface.co/THUDM/chatglm-6b/blob/main/pytorch_model-00001-of-00008.bin

# Download a model by name
aimm install bloom

# Install from an existing amm.json
aimm install
aimm install -r non-default-named.amm.json

# Move packages under aimm management:
aimm mv a b

The API service

No authentication required at the moment. Please play nice.

hashA is the sha256 of the file.

Projects/repos have UUIDs as IDS in AIMM. Civitai IDs are ints (e.g. models/7204), while HuggingFace uses strings (e.g. lmsys/vicuna-7b-delta-v0).

1. Files

1.1 Search files by filename (partial)

https://api.aimm.dev/files?filename=meina&pretty=1
[
    {
        "id": "7b1393c5-e398-4bc4-b416-3f2b3ac24165",
        "hashA": "e03274b1e7478ce7cfd86e4758d9918d26f6f1f73a6e37d425587a7938ac6f79",
        "downloadUrl": "https://civitai.com/api/download/models/16925",
        "filename": "meinamix_meinaV7.safetensors",
        "revision": {
            "id": "c02e9d09-f7cf-49c1-a44e-00a03b0918ed",
            "idInRegistry": "16925",
            "repo": {
                "id": "876abe0f-b774-4629-bcc8-024b2250a374",
                "name": "MeinaMix",
                "registry": "Civitai",
                "idInRegistry": "7240",
                "favour": 102479
            }
        }
    },
    {
        "id": "b4a7b957-f830-4c86-831e-cde04128f767",
        "hashA": "eac6c08a199c4953f80ae5b4ea5f9b2d88c7f3c6f2546e14a57851e3e4a1c5cb",
        "downloadUrl": "https://civitai.com/api/download/models/46137",
        "filename": "meinamix_meinaV9.safetensors",
        "revision": {
            "id": "1f726859-a61a-4df1-9057-20e869f0a54d",
            "idInRegistry": "46137",
            "repo": {
                "id": "876abe0f-b774-4629-bcc8-024b2250a374",
                "name": "MeinaMix",
                "registry": "Civitai",
                "idInRegistry": "7240",
                "favour": 102479
            }
        }
    },
]

Search files by sha256 (must provide full)

License

MIT

Dependencies

~13–28MB
~440K SLoC