Tạo một video!
Đến lượt bạn tạo một video hoàn toàn từ một văn bản gợi ý! Bạn sẽ dùng pipeline CogVideoXPipeline và prompt sau để dẫn hướng quá trình tạo:
Một robot đang nhảy điệu robot. Sàn nhảy có các ô vuông đầy màu sắc và một quả cầu kim tuyến.
Lưu ý: Suy luận trên các mô hình tạo video có thể mất nhiều thời gian, vì vậy chúng tôi đã tải sẵn video đã tạo cho bạn. Chạy prompt khác sẽ không tạo video mới.
Lớp CogVideoXPipeline đã được nhập sẵn cho bạn.
Bài tập này là một phần của khóa học
Mô hình đa phương thức với Hugging Face
Hướng dẫn bài tập
- Tạo một
CogVideoXPipelinetừ checkpointTHUDM/CogVideoX-2b. - Chạy pipeline với prompt đã cho, đặt số bước suy luận là
20, số khung hình cần tạo là20, và guidance scale là6.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
prompt = "A robot doing the robot dance. The dance floor has colorful squares and a glitterball."
# Create a CogVideoXPipeline
pipe = ____(
"____",
torch_dtype=torch.float16
)
# Run the pipeline with the provided prompt
video = pipe(
prompt=____,
num_inference_steps=____,
num_frames=____,
guidance_scale=____
)
video = video.frames[0]
video_path = export_to_video(video, "output.mp4", fps=8)
video = VideoFileClip(video_path)
video.write_gif("video_ex.gif")