Spaces:

Caslow
/

Fortran_to_Rust_Translator

Sleeping

Caslow commited on Nov 21, 2024

Commit

af99375

1 Parent(s): f6dfa4e

cpu

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -34,7 +34,7 @@ def load_model(
     model = AutoModelForCausalLM.from_pretrained(
         pretrained_model_name_or_path=model_name,
-        **kwargs
     )
     model.eval() # Set model to evaluation mode

     model = AutoModelForCausalLM.from_pretrained(
         pretrained_model_name_or_path=model_name,
+        config = kwargs
     )
     model.eval() # Set model to evaluation mode