Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -205,8 +205,7 @@ def main():
|
|
205 |
progress=gr.Progress()
|
206 |
|
207 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2")
|
208 |
-
|
209 |
-
vae = AutoencoderKL.from_single_file("stabilityai/sd-vae-ft-mse-original/vae-ft-mse-840000-ema-pruned.safetensors")
|
210 |
unet = UNet3DConditionModel.to(device, dtype).load_state_dict(load_file(hf_hub_download("ByteDance/SDXL-Lightning", f"sdxl_lightning_{step}step_unet.safetensors"), device=device))
|
211 |
|
212 |
repo = "ByteDance/AnimateDiff-Lightning"
|
|
|
205 |
progress=gr.Progress()
|
206 |
|
207 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2")
|
208 |
+
vae = AutoencoderKL.from_single_file("https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors")
|
|
|
209 |
unet = UNet3DConditionModel.to(device, dtype).load_state_dict(load_file(hf_hub_download("ByteDance/SDXL-Lightning", f"sdxl_lightning_{step}step_unet.safetensors"), device=device))
|
210 |
|
211 |
repo = "ByteDance/AnimateDiff-Lightning"
|