Spaces:

HemaAM
/

AI_Chat_Assistant

Runtime error

HemaAM commited on Dec 22, 2023

Commit

6f58c22

1 Parent(s): e199dfe

bug fix

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -172,7 +172,6 @@ model = AutoModelForCausalLM.from_pretrained(
     "microsoft/phi-2",
     torch_dtype=torch.float32,
     device_map="cpu",
-    device_map="cpu",
     trust_remote_code=True
 )
 model.load_adapter('checkpoint-1100')

     "microsoft/phi-2",
     torch_dtype=torch.float32,
     device_map="cpu",
     trust_remote_code=True
 )
 model.load_adapter('checkpoint-1100')