Hugging Face's logo

Spaces:
abobonbobo13
/
sample
Sleeping

App Files Community
abobonbobo13 commited on
Commit
f99f4fe
·
verified ·
1 Parent(s): 3a5a973

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
4
  MODEL_ID = "rinna/bilingual-gpt-neox-4b-instruction-ppo"
5
  model = AutoModelForCausalLM.from_pretrained(
6
  "rinna/bilingual-gpt-neox-4b-instruction-ppo",
7
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
8
 
9
  device = model.device
10
  device
 
4
  MODEL_ID = "rinna/bilingual-gpt-neox-4b-instruction-ppo"
5
  model = AutoModelForCausalLM.from_pretrained(
6
  "rinna/bilingual-gpt-neox-4b-instruction-ppo",
7
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
8
 
9
  device = model.device
10
  device