Spaces:

Shanat
/

chtbt_tst

Sleeping

Shanat commited on Oct 7

Commit

1f08a45

•

1 Parent(s): 8f5e20e

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("Shanat/llm_tst3")
 model = AutoModelForCausalLM.from_pretrained(
-    "Shanat/llm_tst3",
     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-Instruct")
 model = AutoModelForCausalLM.from_pretrained(
+    "meta-llama/Llama-3.2-1B-Instruct",
     device_map="auto",
     torch_dtype="auto",
 )