Spaces:

piyushgrover
/

PhiGPT

Sleeping

piyushgrover commited on Dec 22, 2023

Commit

fe25480

1 Parent(s): e05d589

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
     model_name,
     low_cpu_mem_usage=True,
     return_dict=True,
-    torch_dtype=torch.float16,
     trust_remote_code=True
     # device_map=device_map,
 )

     model_name,
     low_cpu_mem_usage=True,
     return_dict=True,
+    torch_dtype=torch.float32,
     trust_remote_code=True
     # device_map=device_map,
 )