blabla974 commited on
Commit
1742f2f
·
verified ·
1 Parent(s): 9ff11d3

Fix: switch to stabilityai/stable-video-diffusion-img2vid (public UNetMotionModel-compatible)

Browse files
motion_utils/video_generator_animatediff.py CHANGED
@@ -11,7 +11,7 @@ def generate_video_from_image(image_path, prompt, duration, fps, output_path):
11
  image = load_image(image_path).convert("RGB")
12
 
13
  # Utiliser un modèle AnimateDiff compatible (UNetMotionModel)
14
- model_id = "ali-vilab/animatediff-motion-model"
15
  pipe = AnimateDiffPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
16
  pipe.enable_model_cpu_offload()
17
 
 
11
  image = load_image(image_path).convert("RGB")
12
 
13
  # Utiliser un modèle AnimateDiff compatible (UNetMotionModel)
14
+ model_id = "stabilityai/stable-video-diffusion-img2vid"
15
  pipe = AnimateDiffPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
16
  pipe.enable_model_cpu_offload()
17