Spaces:

MaxKio
/

MioChat

Sleeping

MaxKio commited on 29 days ago

Commit

9703db3

verified ·

1 Parent(s): f63953f

Update hf_Chat.py

Files changed (1) hide show

hf_Chat.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 # ------------------- الإعدادات -------------------
 # غير الاسم إلى أي نموذج تريده على Hugging Face
-MODEL_NAME = "EleutherAI/gpt-neo-2.7B"          # مثال لنموذج مفتوح المصدر
 DEVICE      = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # ------------------- تحميل النموذج ----------------

 # ------------------- الإعدادات -------------------
 # غير الاسم إلى أي نموذج تريده على Hugging Face
+MODEL_NAME = "meta-llama/Llama-2-7b-chat-hf"          # مثال لنموذج مفتوح المصدر
 DEVICE      = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # ------------------- تحميل النموذج ----------------