hep-chat-entrypoint / list_models.py
ho22joshua's picture
Add agent-friendly inference and diagnostics
1375f74
Raw
History Blame Contribute Delete
1.07 kB
import json
from model_config import load_registry
def main():
registry = load_registry()
default = registry.get("default")
for name, config in sorted(registry.get("models", {}).items()):
print(
json.dumps(
{
"name": name,
"default": name == default,
"base_model": config.get("base_model"),
"adapter": config.get("adapter"),
"description": config.get("description", ""),
"parameter_count": config.get("parameter_count"),
"architecture": config.get("architecture"),
"intended_tasks": config.get("intended_tasks", []),
"minimum_requirements": config.get("minimum_requirements", {}),
"recommended_requirements": config.get(
"recommended_requirements", {}
),
},
ensure_ascii=False,
)
)
if __name__ == "__main__":
main()