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.7.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | ed4ea9d9582606a2e9e9e3f54db39dad6f661e6e211bf452bd5b2dc00270190a |
|
MD5 | 344e954648b8ca413c3faae281359f03 |
|
BLAKE2b-256 | 35f2083eab721722c87b576c262aa459bfcf72b5fb915dc440be6980c4bc46a5 |
Hashes for horde_model_reference-0.7.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bd7199ffafc9b510838d9de0ea94187879c7978a7ee6c419e2af12d2d73fbf69 |
|
MD5 | 6789a824b0cf4a1f5c37f95ae71ad53c |
|
BLAKE2b-256 | d4ff314826af7af7b10b9c7952aa605534b4929b525f1effb752e8267729f610 |