tools / get_model_ids.py
patrickvonplaten's picture
up
866074c
raw
history blame
268 Bytes
#!/usr/bin/env python3
from huggingface_hub import HfApi
api = HfApi()
out = api.list_models(filter="diffusers")
model_ids = [x.modelId for x in out]
print(len(model_ids))
with open("./model_ids.txt", "w") as f:
for _id in model_ids:
f.write(_id + "\n")