Spaces:

abdulmalek9
/

Llama_chat

Runtime error

abdulmalek9 commited on Apr 25, 2024

Commit

149a9f3

verified ·

1 Parent(s): bac0c8d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,8 +12,8 @@ def greet(name):
 # Load model directly
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
 # model = "meta-llama/Llama-2-13b-chat-hf" # meta-llama/Llama-2-7b-hf

 # Load model directly
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
 # model = "meta-llama/Llama-2-13b-chat-hf" # meta-llama/Llama-2-7b-hf