Yaron Koresh commited on
Commit
533797d
·
verified ·
1 Parent(s): 624a68a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -205,7 +205,7 @@ def main():
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"
 
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/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"