Update model/utils.py

model/utils.py

@@ -562,8 +562,8 @@ def load_checkpoint(model, ckpt_path, device, use_ema = True):
 
     ckpt_type = ckpt_path.split(".")[-1]
     if ckpt_type == "safetensors":
-       # from safetensors.torch import load_file
-       # checkpoint = load_file(ckpt_path, device=device)
+        from safetensors.torch import load_file
+        checkpoint = load_file(ckpt_path, device=device)
     else:
         checkpoint = torch.load(ckpt_path, weights_only=True, map_location=device)