Yaron Koresh commited on
Commit
f41287b
·
verified ·
1 Parent(s): 17bab31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
24
  base = "black-forest-labs/FLUX.1-dev"
25
 
26
  adapter = MotionAdapter().to(device, dtype)
27
- adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
28
  pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype).to(device)
29
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
30
 
 
24
  base = "black-forest-labs/FLUX.1-dev"
25
 
26
  adapter = MotionAdapter().to(device, dtype)
27
+ adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt, token=os.getenv("hf_token")), device=device))
28
  pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype).to(device)
29
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
30