Spaces:
Running
Running
Fabrice-TIERCELIN
commited on
Commented out the process to find the root cause
Browse files
app.py
CHANGED
@@ -81,6 +81,20 @@ def generate_video_gpu(
|
|
81 |
guidance_scale,
|
82 |
flow_shift,
|
83 |
embedded_guidance_scale
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
84 |
):
|
85 |
print("generate_video_gpu (prompt: " + prompt + ")")
|
86 |
if torch.cuda.device_count() == 0:
|
|
|
81 |
guidance_scale,
|
82 |
flow_shift,
|
83 |
embedded_guidance_scale
|
84 |
+
):
|
85 |
+
return None
|
86 |
+
|
87 |
+
@spaces.GPU(duration=120)
|
88 |
+
def generate_video_gpu2(
|
89 |
+
model,
|
90 |
+
prompt,
|
91 |
+
resolution,
|
92 |
+
video_length,
|
93 |
+
seed,
|
94 |
+
num_inference_steps,
|
95 |
+
guidance_scale,
|
96 |
+
flow_shift,
|
97 |
+
embedded_guidance_scale
|
98 |
):
|
99 |
print("generate_video_gpu (prompt: " + prompt + ")")
|
100 |
if torch.cuda.device_count() == 0:
|