Spaces:

codeparrot
/

codegen-subspace

Runtime error

loubnabnl HF staff commited on Aug 25, 2022

Commit

2e1fac1

1 Parent(s): c785e1b

use 2B checkpoint instead of 6B

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,8 +10,8 @@ example = [
     ["def get_file_size(filepath):", 24, 0.6, 42],
     ["def count_lines(filename):", 40, 0.6, 42],
     ["def count_words(filename):", 40, 0.6, 42]]
-tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-6B-mono")
-model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-6B-mono")
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):

     ["def get_file_size(filepath):", 24, 0.6, 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):