antimoda1 commited on
Commit ·
49208b1
1
Parent(s): 04a717b
make util
Browse files
utils/possible_llm_models.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Данная утилита получает список моделей с Hugging Face, отфильтрованных по ключевому слову "chat", которые можно использовать для чат-бота"""
|
| 2 |
+
|
| 3 |
+
from huggingface_hub import HfApi
|
| 4 |
+
|
| 5 |
+
api = HfApi()
|
| 6 |
+
models = api.list_models(filter="chat", limit=10)
|
| 7 |
+
|
| 8 |
+
for m in models:
|
| 9 |
+
print(m.modelId)
|