Yaron Koresh commited on
Commit
74640ef
·
verified ·
1 Parent(s): f41287b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ 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, 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
 
31
  def translate(text,lang):
 
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, token=os.getenv("hf_token")).to(device)
29
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
30
 
31
  def translate(text,lang):