Skip to content

Getting Started

AIMM is an app to manage your machine learning models

๐Ÿ“ Description

AI Model Manager CLI, it's like pip or npm - but for AI models!

For users, this allows easily managing your collection of AI models. It stores your models in a location that all of your AI enabled apps can access, so you don't need to worry about the huge files taking up so much space, or creating symlinks. Finally, no more having to hunt for a model and making sure you rename it and put it in the correct folder correctly.

For developers, it means less code that you have to handle initializing your app and collecting all of the AI model resources needed. AI Model Manager also will provide security through checksum checks and pickle scanning.

๐Ÿ“ฆ Installation

From Download, download the latest version of AIMM for your operating system.

๐Ÿ’ป Quickstart

Basic Use

๐Ÿ“ Project Commands

๐ŸŽฌ init

Initialize a project

First we have to initialize the aimodels.json file

aimm init

Example: Basic Use

โž• add

Add a model to your project (aimodels.json)

aimm add [model-name]:[version]

Then run the add command to add the model to the aimodels.json file. This will also download the model to the models folder.

Basic Use

โž– remove

Remove a model from project (aimodels.json)

aimm remove [model-name]:[version]

Basic Use

๐Ÿ“ System Commands

Search for a model

aimm search BSRGAN

Second search for a model you want to use. For example the BSRGAN model, a super resolution model that can upscale images.

Basic Use

๐Ÿงพ list

List all installed models (system wide)

aimm list

Basic Use

๐ŸŽˆ install

Install a model to the system

aimm install [model-name]:[version]

Basic Use

๐Ÿ—‘ uninstall

Uninstall a model from the system

aimm uninstall [model-name]:[version]

Basic Use

๐Ÿ“‡ info

Get info about a model

aimm info [model-name]:[version]

Basic Use

๐Ÿ“ก scan

Scan a model for pickles / malicious code

aimm scan [model-name]:[version]

Basic Use

๐Ÿ”ฎ check-updates

Check for updates

aimm --check-updates

Basic Use

๐Ÿ“œ license

View License Information

aimm --version  [model-name]:[version]
aimm --version -verbose [model-name]:[version]

Basic Use

๐Ÿ“œ License

Apache License 2.0