Yaron Koresh commited on
Commit
c5d0af0
·
verified ·
1 Parent(s): d07943a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -196,7 +196,7 @@ ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
196
  base = "emilianJR/epiCRealism"
197
  #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
198
 
199
- pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, feature_extractor=None, image_encoder=None, unet=unet, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
200
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
201
  pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
202
  pipe.enable_free_init(method="butterworth", use_fast_sampling=False)
 
196
  base = "emilianJR/epiCRealism"
197
  #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
198
 
199
+ pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
200
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
201
  pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
202
  pipe.enable_free_init(method="butterworth", use_fast_sampling=False)