Upload convert_repo_to_safetensors_sd_gr.py

convert_repo_to_safetensors_sd_gr.py

@@ -339,6 +339,7 @@ def convert_diffusers_to_safetensors(model_path, checkpoint_path, dtype="fp16",
     if dtype == "fp16": state_dict = {k: v.half() for k, v in state_dict.items()}
     elif dtype == "fp32": state_dict = {k: v.to(torch.float32) for k, v in state_dict.items()}
     elif dtype == "bf16": state_dict = {k: v.to(torch.bfloat16) for k, v in state_dict.items()}
+    elif dtype == "fp8": state_dict = {k: v.to(torch.float8_e4m3fn) for k, v in state_dict.items()}
 
     save_file(state_dict, checkpoint_path)
 
@@ -417,7 +418,7 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
 
     parser.add_argument("--repo_id", default=None, type=str, required=True, help="HF Repo ID of the model to convert.")
-    parser.add_argument("--dtype", default="fp16", type=str, choices=["fp16", "fp32", "bf16", "default"], help='Output data type. (Default: "fp16")')
+    parser.add_argument("--dtype", default="fp16", type=str, choices=["fp16", "fp32", "bf16", "fp8", "default"], help='Output data type. (Default: "fp16")')
 
     args = parser.parse_args()
     assert args.repo_id is not None, "Must provide a Repo ID!"