LIFineTuned

Paused

alexkueck commited on Jun 21, 2023

Commit

9f21ce0

1 Parent(s): 6dc4661

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -81,7 +81,7 @@ def load_tokenizer_and_model(base_model, use_auth_token, load_8bit=False):
             load_in_8bit=load_8bit,
             torch_dtype=torch.float16,
             device_map="auto",
-            use_auth_token
         )
     else:
         model = AutoModelForCausalLM.from_pretrained(

             load_in_8bit=load_8bit,
             torch_dtype=torch.float16,
             device_map="auto",
+            use_auth_token=use_auth_token
         )
     else:
         model = AutoModelForCausalLM.from_pretrained(