A helper library providing a way to work with the lists of diffusion models, utility models, and any other related files required for AI-Horde.
Project description
Horde Model Reference
This package provides some tools to help manage the models which power the AI-Horde.
Reference info (.json files)
For now, the legacy reference format will be availible as before at the original official repo used in the past.
stable_diffusion.json
You can find a schema for an individual record in the file stable_diffusion.schema.json
in the root of this repository. Also see stable_diffusion.example.json
for a small example containing two records will all of the fields populated.
stable_diffusion.json changes
You can see two records which include entries for every field, and the associated metadata in stable_diffusion.example.json
at the root of this repository.
Some key takeaways for the new stable_diffusion.json
:
- The following keys have been removed:
type
download_all
available
- the sub-key
file_path
underdownload
- the entire key
files
underconfig
has been removed.- (
config
still contains adownload
key, which is a list of all files to download.)
- (
baseline
's old values have been normalized. The currently valid values are as follows:stable_diffusion_1
stable_diffusion_2_768
stable_diffusion_2_512
stable_diffusion_xl
stable_cascade
- An MD5 sum is no longer included. All models (of all types) will have an SHA included from now on.
download
entries optionally contain a new key,known_slow_download
, which indicates this download host is known to be slow at times.
Moving forward, you can expect the schema to honor at least the existing values. There is a strong possibility additional fields will be added.
Python library
This repo is also a python library designed to help you integrate the scheme the AI-Horde project uses to manage its models into your project.
General info
You can install this module through pip:
python -m pip install horde_model_reference
This library has a number of python classes which may assist you in working with the model reference. The following files may be of interest:
- horde_model_reference\model_reference_records.py
- Contains pydantic definitions, and some meta information, for all record types.
- horde_model_reference\meta_consts.py
- Contains many commonly used strings, enums, and certain useful dict lookups.
- horde_model_reference\path_consts.py
- Contains certain potentially useful paths, path constructors and folder/file name information relevant to the package.
Note that a number of useful imports have been made availible at the horde_model_reference
import level.
Horde Moderators/Support Staff
Validating
When making changes to stable_diffusion.json, you can now validate, format, and standardize it for consistency. You can do this by invoking the following:
validate_sd stable_diffusion.json
This will give you a success message if the file is standardized. If it is not, you can invoke the following:
validate_sd stable_diffusion.json --write validated_and_formatted.json
This will write the appropriately normalized json out to the path specified by --write
. It will only write a file out if the input file is valid json and conforms to the established schema.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for horde_model_reference-0.6.3.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 115ec0987a4a4108a23c342ef29ebf01d3ef0fa13b672d13fdd811e7ad7ad9df |
|
MD5 | a6773b6346b7332becebdd96f033aa17 |
|
BLAKE2b-256 | 985bf5f6d25574d368e71c415a1ea79f7c2636c98ccce99eaaa5613a62a6cd66 |
Hashes for horde_model_reference-0.6.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5e0c0a4a319f8ad80686b2d493edb898c28aafb2c2770095e32f3c700ffa1d3b |
|
MD5 | ae0a481e77272248a230aacac2cf99ac |
|
BLAKE2b-256 | 0679393ab951188cd352168e08800e73d96fc1bcbabb9017b9c6b1cadbe350f9 |