Spaces:

valencar
/

chat-yi

Runtime error

valencar commited on Aug 10, 2024

Commit

b30e5eb

1 Parent(s): 496d539

fdfsdfs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,8 +17,8 @@ before = datetime.datetime.now()
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
-tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-34B-Chat-4bits")
-model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-34B-Chat-4bits")
 st.write('tokenizando...')
 prompt = "Hey, are you conscious? Can you talk to me?"

 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-1.5-9B")
+model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-1.5-9B")
 st.write('tokenizando...')
 prompt = "Hey, are you conscious? Can you talk to me?"