Spaces:

Shanat
/

chtbt_tst

Sleeping

Shanat commited on Oct 7

Commit

5d10772

•

1 Parent(s): 1f08a45

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,9 +18,9 @@ import torch
 login(token = os.getenv('HF_TOKEN'))
 #chatbot = pipeline(model="meta-llama/Llama-3.2-1B")
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
 model = AutoModelForCausalLM.from_pretrained(
-    "meta-llama/Llama-3.2-1B-Instruct",
     device_map="auto",
     torch_dtype="auto",
 )

 login(token = os.getenv('HF_TOKEN'))
 #chatbot = pipeline(model="meta-llama/Llama-3.2-1B")
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
 model = AutoModelForCausalLM.from_pretrained(
+    "meta-llama/Llama-3.2-1B",
     device_map="auto",
     torch_dtype="auto",
 )