mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
update
This commit is contained in:
@@ -0,0 +1,17 @@
|
||||
import sys
|
||||
import huggingface_hub as hf
|
||||
from rich import print # pylint: disable=redefined-builtin
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.argv.pop(0)
|
||||
keyword = sys.argv[0] if len(sys.argv) > 0 else ''
|
||||
hf_api = hf.HfApi()
|
||||
model_filter = hf.ModelFilter(
|
||||
model_name=keyword,
|
||||
task='text-to-image',
|
||||
tags='stable-diffusion',
|
||||
library=['diffusers', 'stable-diffusion'],
|
||||
)
|
||||
res = hf_api.list_models(filter=model_filter, full=True, limit=50, sort="downloads", direction=-1)
|
||||
models = [{ 'name': m.modelId, 'downloads': m.downloads, 'mtime': m.lastModified, 'url': f'https://huggingface.co/{m.modelId}' } for m in res]
|
||||
print('Online', models)
|
||||
Reference in New Issue
Block a user