Update merged_files3.py

merged_files3.py

@@ -495,7 +495,7 @@ dtype = torch.float16  # Use float16 consistently for all models
 sd_offset = sf.load_file(model_path) # Use device variable
 sd_origin = unet.state_dict()
 keys = sd_origin.keys()
-sd_offset = {k: v.to(device) for k, v in sd_offset.items()} # Move each tensor to GPUunet.load_state_dict(sd_merged, strict=True)
+sd_merged = {k: v.to(device) for k, v in sd_offset.items()} # Move each tensor to GPUunet.load_state_dict(sd_merged, strict=True)
 del sd_offset, sd_origin, sd_merged, keys