envs

app.py

@@ -82,14 +82,28 @@ If you have any questions, please feel free to reach me out at <b>ywl@stu.pku.ed
 
 os.makedirs("models/personalized")
 os.makedirs("models/sd1-5")
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/blob/main/flow_controlnet.ckpt -P models/')
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/blob/main/image_controlnet.ckpt -P models/')
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/blob/main/unet.ckpt -P models/')
-os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/unet/config.json -P models/sd1-5/')
-os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/unet/diffusion_pytorch_model.bin -P models/sd1-5/')
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/blob/main/helloobjects_V12c.safetensors -P models/personalized')
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/blob/main/TUSUN.safetensors -P models/personalized')
+os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/flow_controlnet.ckpt?download=true -P models/')
+os.system(f'mv models/flow_controlnet.ckpt?download=true models/flow_controlnet.ckpt')
 
+os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/image_controlnet.ckpt?download=true -P models/')
+os.system(f'mv models/image_controlnet.ckpt?download=true models/image_controlnet.ckpt')
+
+os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/unet.ckpt?download=true -P models/')
+os.system(f'mv models/unet.ckpt?download=true models/unet.ckpt')
+
+
+os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/unet/config.json?download=true -P models/sd1-5/')
+os.system(f'mv models/sd1-5/config.json?download=true  models/sd1-5/config.json')
+
+os.system(f'cp -r models/unet.ckpt  models/sd1-5/unet.ckpt')
+
+# os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/unet/diffusion_pytorch_model.bin?download=true -P models/sd1-5/')
+
+os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/helloobjects_V12c.safetensors?download=true -P models/personalized')
+os.system(f'mv models/personalized/helloobjects_V12c.safetensors?download=true models/personalized/helloobjects_V12c.safetensors')
+
+os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/TUSUN.safetensors?download=true -P models/personalized')
+os.system(f'mv models/personalized/TUSUN.safetensors?download=true models/personalized/TUSUN.safetensors')
 
 
 

modules/unet.py

@@ -565,9 +565,9 @@ class UNet3DConditionFlowModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMi
             "CrossAttnUpBlock3D"
         ]
 
-        from diffusers.utils import WEIGHTS_NAME
+        # from diffusers.utils import WEIGHTS_NAME
         model = cls.from_config(config, **unet_additional_kwargs)
-        model_file = os.path.join(pretrained_model_path, WEIGHTS_NAME)
+        model_file = os.path.join(pretrained_model_path, "unet.ckpt")
         if not os.path.isfile(model_file):
             raise RuntimeError(f"{model_file} does not exist")
         state_dict = torch.load(model_file, map_location="cpu")