Update model_converter.py

model_converter.py

@@ -4,7 +4,7 @@ import safetensors.torch
 def load_from_standard_weights(input_file: str, device: str) -> dict[str, torch.Tensor]:
     # Taken from: https://github.com/kjsman/stable-diffusion-pytorch/issues/7#issuecomment-1426839447
     # original_model = torch.load(input_file, map_location=device, weights_only = False)["state_dict"]
-    original_model=torch.load(input_file, weights_only = False)["state_dict"]
+    original_model=torch.load(input_file, map_location=device ,weights_only = False)["state_dict"]
 
     converted = {}
     converted['diffusion'] = {}