Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ repo = "ByteDance/AnimateDiff-Lightning"
|
|
36 |
ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
37 |
#base = "emilianJR/epiCRealism"
|
38 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
39 |
-
|
40 |
#unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
|
41 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
42 |
|
|
|
36 |
ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
37 |
#base = "emilianJR/epiCRealism"
|
38 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
39 |
+
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
|
40 |
#unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
|
41 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
42 |
|