support diffusers ms models
This commit is contained in:
13
diffusers_video.py
Normal file
13
diffusers_video.py
Normal 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
|
||||
Reference in New Issue
Block a user