Skip to content

Haidra-Org/horde-model-reference

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

411 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Horde Model Reference

Haidra Logo

Build Status PyPI Version Supported Python versions License: AGPL v3

Horde Model Reference is the authoritative source for AI model metadata in the AI-Horde ecosystem: download URLs, SHA-256 checksums, baselines, NSFW flags, and capabilities for image generation, text generation, and utility models used by workers, clients, and services.

It is three things in one package: a set of JSON reference files, a Python library, and a FastAPI service.

Install

pip install horde-model-reference

Quick try

from horde_model_reference import ModelReferenceManager, MODEL_REFERENCE_CATEGORY

manager = ModelReferenceManager()  # auto-fetches from the PRIMARY server, caches in memory

models = manager.get_model_reference(MODEL_REFERENCE_CATEGORY.image_generation)
for name, model in models.items():
    print(f"{name}: {model.baseline}, NSFW={model.nsfw}")

No configuration needed -- the defaults work out of the box. The library runs in read-only REPLICA mode, fetching from models.aihorde.net with automatic fallback to the original GitHub repositories if the PRIMARY server is unreachable.

For the fluent query API (filter, sort, aggregate, merge sources), HTTP API access, deploying your own server, or contributing model data, see the full documentation.

Documentation

Full docs at horde-model-reference.readthedocs.io/en/latest/. Build locally with mkdocs serve.

Runnable examples live in the examples/ directory.

Contributing

Contributions are welcome. Read CONTRIBUTING.md for setup and guidelines, and see the open issues.

Community

License

Licensed under the GNU Affero General Public License v3.0 (AGPL-3.0). See LICENSE for the full text. If you run a modified version as a network service, you must make your source code available to users. See the GNU AGPL FAQ for details.

Related Projects

About

This package provides some tools to help manage the model metadata which AI-Horde uses

Resources

License

Contributing

Stars

Watchers

Forks

Sponsor this project

 

Packages

 
 
 

Contributors