Spaces:

xset
/

aibot

Sleeping

xset commited on May 18

Commit

85e9a75

1 Parent(s): 3908fe1

TinyLlama/TinyLlama_v1.1

Files changed (1) hide show

main.py CHANGED Viewed

@@ -16,7 +16,7 @@ class RequestData(BaseModel):
 #MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 #MODEL_NAME = "ai-forever/rugpt3small_based_on_gpt2"
-MODEL_NAME = "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF"
 try:
     # Загрузка модели с явным указанием device_map

 #MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 #MODEL_NAME = "ai-forever/rugpt3small_based_on_gpt2"
+MODEL_NAME = "TinyLlama/TinyLlama_v1.1"
 try:
     # Загрузка модели с явным указанием device_map