Spaces:

codeparrot
/

codeparrot-subspace

Runtime error

loubnabnl HF Staff commited on May 23, 2022

Commit

41612f9

1 Parent(s): 43385c3

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ example = [
     ["\"""import GPT2 from transformers\""" ", "Sample", 16, 42],
     ["def get_file_size(filepath):", "Sample", 8, 42]]
 tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
-model = AutoModelForCausalLM.from_pretrained("lvwerra/codeparrot", low_cpu_mem_usage)
 def code_generation(gen_prompt, strategy, max_tokens, seed=42):

     ["\"""import GPT2 from transformers\""" ", "Sample", 16, 42],
     ["def get_file_size(filepath):", "Sample", 8, 42]]
 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):