Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
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
|
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):
|