Hugging Face's logo

Spaces:
valencar
/
chat-yi
Runtime error

App Files Community
valencar commited on
Commit
885a80a
·
1 Parent(s): b30e5eb

fdfsdfs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ before = datetime.datetime.now()
17
  # Load model directly
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
- tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-1.5-9B")
21
- model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-1.5-9B")
22
 
23
  st.write('tokenizando...')
24
  prompt = "Hey, are you conscious? Can you talk to me?"
 
17
  # Load model directly
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
+ tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-6B-200K")
21
+ model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-6B-200K")
22
 
23
  st.write('tokenizando...')
24
  prompt = "Hey, are you conscious? Can you talk to me?"