mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
24 lines
758 B
Python
24 lines
758 B
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
'''
|
||
|
|
This script is used at release time to generate a markdown table describing the
|
||
|
|
starter models. This text is then manually copied into 050_INSTALL_MODELS.md.
|
||
|
|
'''
|
||
|
|
|
||
|
|
from omegaconf import OmegaConf
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
initial_models_file = Path(__file__).parent / '../invokeai/configs/INITIAL_MODELS.yaml'
|
||
|
|
models = OmegaConf.load(initial_models_file)
|
||
|
|
print('|Model Name | HuggingFace Repo ID | Description | URL |')
|
||
|
|
print('|---------- | ---------- | ----------- | --- |')
|
||
|
|
for model in models:
|
||
|
|
repo_id = models[model].repo_id
|
||
|
|
url = f'https://huggingface.co/{repo_id}'
|
||
|
|
print(f'|{model}|{repo_id}|{models[model].description}|{url} |')
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|