Yaron Koresh commited on
Commit
066d78b
·
verified ·
1 Parent(s): 4247ff9

Update app.py

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