rajux75 commited on
Commit
c1f1913
Β·
verified Β·
1 Parent(s): 3a03ef4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ print("Stable Diffusion 1.5 model loaded.")
41
  # Load Animation Model (AnimateDiff)
42
  print("Loading Animation Model (AnimateDiff)...")
43
  # Load motion module
44
- adapter = MotionAdapter.from_pretrained("emperorxi/animatediff-motion-module-v1", torch_dtype=torch.float16 if device == "cuda" else torch.float32)
45
  # Load base SD pipeline
46
  pipe_anim = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", motion_adapter=adapter, torch_dtype=torch.float16 if device == "cuda" else torch.float32).to(device)
47
  # Configure scheduler
 
41
  # Load Animation Model (AnimateDiff)
42
  print("Loading Animation Model (AnimateDiff)...")
43
  # Load motion module
44
+ adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5", torch_dtype=torch.float16 if device == "cuda" else torch.float32)
45
  # Load base SD pipeline
46
  pipe_anim = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", motion_adapter=adapter, torch_dtype=torch.float16 if device == "cuda" else torch.float32).to(device)
47
  # Configure scheduler