Update sh.py

sh.py

@@ -2,8 +2,8 @@ import torch
 from safetensors.torch import load_file
 
 # Load both models
-model1 = load_file('merged_model0.safetensors')
-model2 = load_file('diffusion_pytorch_model-00001-of-00003.safetensors')
+model1 = load_file('merged_model.safetensors')
+model2 = load_file('diffusion_pytorch_model-00002-of-00003.safetensors')
 
 # Open log.txt for writing the output
 with open('log.txt', 'w') as log_file: