multimodalart HF staff commited on
Commit
37393f2
1 Parent(s): 3636642

Update convertosd.py

Browse files
Files changed (1) hide show
  1. convertosd.py +1 -6
convertosd.py CHANGED
@@ -195,12 +195,7 @@ def convert_text_enc_state_dict(text_enc_dict):
195
  return text_enc_dict
196
 
197
 
198
- if __name__ == "__main__":
199
-
200
-
201
- model_path = ""
202
- checkpoint_path= ""
203
-
204
  unet_path = osp.join(model_path, "unet", "diffusion_pytorch_model.bin")
205
  vae_path = osp.join(model_path, "vae", "diffusion_pytorch_model.bin")
206
  text_enc_path = osp.join(model_path, "text_encoder", "pytorch_model.bin")
 
195
  return text_enc_dict
196
 
197
 
198
+ def convert(model_path, checkpoint_path):
 
 
 
 
 
199
  unet_path = osp.join(model_path, "unet", "diffusion_pytorch_model.bin")
200
  vae_path = osp.join(model_path, "vae", "diffusion_pytorch_model.bin")
201
  text_enc_path = osp.join(model_path, "text_encoder", "pytorch_model.bin")