Update app.py

app.py

@@ -200,7 +200,7 @@ def main():
     fps=20
     time=16
     device = "cuda"
-    dtype = torch.float16
+    dtype = torch.bfloat16
     result=[]
     step = 2
 
@@ -218,7 +218,7 @@ def main():
     repo = "ByteDance/AnimateDiff-Lightning"
     ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 
-    pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype).to(device)
+    pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
     pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)