Yaron Koresh commited on
Commit
10da314
·
verified ·
1 Parent(s): 59f1e4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ step = 2
33
  #ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
34
  #base = "emilianJR/epiCRealism"
35
  base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
36
- vae = AutoencoderKL.from_single_file("https://huggingface.co/emilianJR/epiCRealism/vae/diffusion_pytorch_model.safetensors").to(device, dtype=dtype)
37
  #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)
38
  adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
39
 
 
33
  #ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
34
  #base = "emilianJR/epiCRealism"
35
  base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
36
+ vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
37
  #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)
38
  adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
39