Spaces:

codeparrot
/

codegen-subspace

Runtime error

loubnabnl HF Staff commited on May 27, 2022

Commit

a4a50f8

1 Parent(s): c8aa726

update description

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ example = [
     ["def count_lines(filename):", 40, 0.6, 42],
     ["def count_words(filename):", 40, 0.6, 42]]
 tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-2B-mono")
-model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-2B-mono", low_cpu_mem_usage=True)
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):

     ["def count_lines(filename):", 40, 0.6, 42],
     ["def count_words(filename):", 40, 0.6, 42]]
 tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-2B-mono")
+model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-2B-mono")
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):