support diffusers ms models

This commit is contained in:
2025-09-08 16:32:50 +08:00
parent 0ae0bd7904
commit 495c3fcd8a
6 changed files with 47 additions and 7 deletions

13
diffusers_video.py Normal file
View File

@@ -0,0 +1,13 @@
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import export_to_video
model_path = "/mnt/models/AI-ModelScope/text-to-video-ms-1.7b"
pipe = DiffusionPipeline.from_pretrained(
model_path, torch_dtype=torch.float16, variant="fp16"
)
pipe.enable_model_cpu_offload() # 省显存
pipe.enable_vae_slicing()
frames = pipe("Spiderman is surfing", num_frames=16).frames[0]
export_to_video(frames, "output.mp4") # 默认约2秒/8fps