Spaces:
Sleeping
Sleeping
Priyanshi Saxena
commited on
Commit
·
3f8cfad
1
Parent(s):
9451ca9
Fix dependency versions and improve error handling
Browse files- requirements.txt +4 -3
- src/video_generator.py +3 -1
requirements.txt
CHANGED
@@ -2,9 +2,10 @@ gradio==4.44.0
|
|
2 |
torch==2.1.0
|
3 |
torchvision==0.16.0
|
4 |
torchaudio==2.1.0
|
5 |
-
diffusers==0.
|
6 |
-
transformers==4.
|
7 |
-
accelerate==0.
|
|
|
8 |
opencv-python==4.8.1.78
|
9 |
Pillow==10.1.0
|
10 |
moviepy==1.0.3
|
|
|
2 |
torch==2.1.0
|
3 |
torchvision==0.16.0
|
4 |
torchaudio==2.1.0
|
5 |
+
diffusers==0.30.0
|
6 |
+
transformers==4.44.0
|
7 |
+
accelerate==0.34.0
|
8 |
+
huggingface_hub==0.24.0
|
9 |
opencv-python==4.8.1.78
|
10 |
Pillow==10.1.0
|
11 |
moviepy==1.0.3
|
src/video_generator.py
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
import torch
|
2 |
-
from diffusers import StableVideoDiffusionPipeline
|
3 |
from PIL import Image
|
4 |
import numpy as np
|
5 |
from config.settings import DEVICE, MODEL_CONFIG, PROCESSING_CONFIG
|
@@ -12,6 +11,7 @@ class VideoGenerator:
|
|
12 |
def _load_models(self):
|
13 |
if DEVICE == "cuda":
|
14 |
try:
|
|
|
15 |
self.pipeline = StableVideoDiffusionPipeline.from_pretrained(
|
16 |
MODEL_CONFIG['svd_model'],
|
17 |
torch_dtype=torch.float16,
|
@@ -21,9 +21,11 @@ class VideoGenerator:
|
|
21 |
self.pipeline.enable_model_cpu_offload()
|
22 |
except Exception as e:
|
23 |
print(f"Failed to load SVD: {e}")
|
|
|
24 |
|
25 |
def generate_video(self, image, prompt, duration=2):
|
26 |
if not self.pipeline:
|
|
|
27 |
return None
|
28 |
|
29 |
try:
|
|
|
1 |
import torch
|
|
|
2 |
from PIL import Image
|
3 |
import numpy as np
|
4 |
from config.settings import DEVICE, MODEL_CONFIG, PROCESSING_CONFIG
|
|
|
11 |
def _load_models(self):
|
12 |
if DEVICE == "cuda":
|
13 |
try:
|
14 |
+
from diffusers import StableVideoDiffusionPipeline
|
15 |
self.pipeline = StableVideoDiffusionPipeline.from_pretrained(
|
16 |
MODEL_CONFIG['svd_model'],
|
17 |
torch_dtype=torch.float16,
|
|
|
21 |
self.pipeline.enable_model_cpu_offload()
|
22 |
except Exception as e:
|
23 |
print(f"Failed to load SVD: {e}")
|
24 |
+
self.pipeline = None
|
25 |
|
26 |
def generate_video(self, image, prompt, duration=2):
|
27 |
if not self.pipeline:
|
28 |
+
print("Pipeline not available, skipping video generation")
|
29 |
return None
|
30 |
|
31 |
try:
|