修正了一点小错误
Browse files- .gitignore +1 -0
- generate.py +19 -0
.gitignore
CHANGED
@@ -1,3 +1,4 @@
|
|
1 |
*.bin
|
2 |
*.safetensors
|
3 |
/.idea
|
|
|
|
1 |
*.bin
|
2 |
*.safetensors
|
3 |
/.idea
|
4 |
+
/output_videos
|
generate.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import numpy as np
|
2 |
+
import torch
|
3 |
+
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
4 |
+
from diffusers.utils import export_to_video
|
5 |
+
from datetime import datetime
|
6 |
+
|
7 |
+
pipe = DiffusionPipeline.from_pretrained(r"J:\Projects\Video-Projects\text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
|
8 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
9 |
+
pipe.enable_model_cpu_offload()
|
10 |
+
timestamp_str = datetime.now().strftime("%Y-%m-%d-%H%M%S")
|
11 |
+
|
12 |
+
output_video_path=f"J:/Projects/Video-Projects/text-to-video-ms-1.7b/output_videos/{timestamp_str}.mp4"
|
13 |
+
prompt = "Spiderman is surfing"
|
14 |
+
|
15 |
+
video_frames = pipe(prompt, num_inference_steps=25).frames
|
16 |
+
video_frames_np = [np.array(frame) for frame in video_frames]
|
17 |
+
video_frames_np = np.concatenate(video_frames_np, axis=0)
|
18 |
+
|
19 |
+
video_path = export_to_video(video_frames_np,output_video_path)
|