Faizbulbul commited on
Commit
f0fe9b3
·
verified ·
1 Parent(s): cb78791

Delete Video_generator.py

Browse files
Files changed (1) hide show
  1. Video_generator.py +0 -14
Video_generator.py DELETED
@@ -1,14 +0,0 @@
1
- from diffusers import StableVideoDiffusionPipeline
2
- import torch
3
-
4
- # Model Load
5
- pipe = StableVideoDiffusionPipeline.from_pretrained(
6
- "stabilityai/stable-video-diffusion-img2vid",
7
- torch_dtype=torch.float16
8
- ).to("cuda")
9
-
10
- def generate_video(prompt):
11
- video = pipe(prompt, num_inference_steps=25).frames
12
- video_path = "generated_video.mp4"
13
- video[0].save(video_path) # Save the first generated video
14
- return video_path