Spaces:

dar-tau
/

speaking-probes

Runtime error

guy-dar commited on Dec 28, 2022

Commit

1ade477

1 Parent(s): 3261bf5

minor fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ def load_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.pad_token = tokenizer.eos_token
     model = AutoModelForCausalLM.from_pretrained(model_name)
-    model_params = extract_gpt_parameters(model_name)
     return model, model_params, tokenizer

     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.pad_token = tokenizer.eos_token
     model = AutoModelForCausalLM.from_pretrained(model_name)
+    model_params = extract_gpt_parameters(model)
     return model, model_params, tokenizer