Spaces:

codeparrot
/

codeparrot-subspace

Runtime error

loubnabnl HF Staff commited on May 23, 2022

Commit

5cea6a1

1 Parent(s): f3048cf

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ title = "CodeParrot Generator"
 description = "This is a subspace to make code generation with CodeParrot, it is used in a larger space for model comparison."
 example = "def print_hello_world():\n \"""Print 'Hello world' \""" "
 tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
-model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_usage=True)
 def code_generation(gen_prompt, strategy, max_tokens, seed=42):

 description = "This is a subspace to make code generation with CodeParrot, it is used in a larger space for model comparison."
 example = "def print_hello_world():\n \"""Print 'Hello world' \""" "
 tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
+model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot")
 def code_generation(gen_prompt, strategy, max_tokens, seed=42):