Spaces:

flax-community
/

code-clippy-problem-solver

Runtime error

arampacha commited on Jul 19, 2021

Commit

ced24dc

1 Parent(s): 8036522

tok

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_name = "flax-community/gpt-code-clippy-1.3B-apps-alldata-2"
 model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
 tokenizer.pad_token = tokenizer.eos_token

 model_name = "flax-community/gpt-code-clippy-1.3B-apps-alldata-2"
 model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 tokenizer.pad_token = tokenizer.eos_token