Spaces:

sanaweb
/

PartAI-Dorna-Llama3-8B-Instruct

Sleeping

sanaweb commited on Jul 8

Commit

f8d903d

•

1 Parent(s): becd822

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ if not torch.cuda.is_available():
 if torch.cuda.is_available():
-    model_id = "PartAI/Dorna-Llama3-8B-Instruct"
     model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16)
     tokenizer = AutoTokenizer.from_pretrained(model_id)

 if torch.cuda.is_available():
+    model_id = "https://huggingface.co/PartAI/Dorna-Llama3-8B-Instruct"
     model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16)
     tokenizer = AutoTokenizer.from_pretrained(model_id)