Update libra/model/builder.py

libra/model/builder.py

@@ -30,10 +30,6 @@ def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, l
     if device != "cuda":
         kwargs['device_map'] = {"": device}
         
-    if device == "cpu":
-        kwargs["torch_dtype"] = torch.float32
-    else:
-        kwargs["torch_dtype"] = torch.float16
         
     if load_8bit:
         kwargs['load_in_8bit'] = True