Upload 2 files

convert_repo_to_safetensors_sd.py

@@ -227,15 +227,12 @@ def convert_diffusers_to_safetensors(model_path, checkpoint_path, half = True):
     text_enc_dict = {"cond_stage_model.transformer." + k: v for k, v in text_enc_dict.items()}
 
     # Put together new checkpoint
+    state_dict = {**unet_state_dict, **vae_state_dict, **text_enc_dict}
+    if half:
+        state_dict = {k:v.half() for k,v in state_dict.items()}
     if input_safetensors:
-        state_dict = {**unet_state_dict, **vae_state_dict, **text_enc_dict}
-        if half:
-            state_dict = {k:v.half() for k,v in state_dict.items()}
-        save_file(state_dict, checkpoint_path, metadata={"format": "pt"})
+        save_file(state_dict, checkpoint_path)
     else:
-        state_dict = {**unet_state_dict, **vae_state_dict, **text_enc_dict}
-        if half:
-            state_dict = {k:v.half() for k,v in state_dict.items()}
         state_dict = {"state_dict": state_dict}
         torch.save(state_dict, checkpoint_path)
 

convert_repo_to_safetensors_sd_gr.py

@@ -228,15 +228,12 @@ def convert_diffusers_to_safetensors(model_path, checkpoint_path, half = True, p
     text_enc_dict = {"cond_stage_model.transformer." + k: v for k, v in text_enc_dict.items()}
 
     # Put together new checkpoint
+    state_dict = {**unet_state_dict, **vae_state_dict, **text_enc_dict}
+    if half:
+        state_dict = {k:v.half() for k,v in state_dict.items()}
     if input_safetensors:
-        state_dict = {**unet_state_dict, **vae_state_dict, **text_enc_dict}
-        if half:
-            state_dict = {k:v.half() for k,v in state_dict.items()}
-        save_file(state_dict, checkpoint_path, metadata={"format": "pt"})
+        save_file(state_dict, checkpoint_path)
     else:
-        state_dict = {**unet_state_dict, **vae_state_dict, **text_enc_dict}
-        if half:
-            state_dict = {k:v.half() for k,v in state_dict.items()}
         state_dict = {"state_dict": state_dict}
         torch.save(state_dict, checkpoint_path)