AIMM is an app to manage your machine learning models
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.
From Download, download the latest version of AIMM for your operating system.
📝 Project Commands
Initialize a project
First we have to initialize the aimodels.json file
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.
Remove a model from project (aimodels.json)
aimm remove [model-name]:[version]
📝 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.
List all installed models (system wide)
Install a model to the system
aimm install [model-name]:[version]
Uninstall a model from the system
aimm uninstall [model-name]:[version]
Get info about a model
aimm info [model-name]:[version]
Scan a model for pickles / malicious code
aimm scan [model-name]:[version]
Check for updates
View License Information
aimm --version [model-name]:[version]
aimm --version -verbose [model-name]:[version]
Apache License 2.0