Fabrice-TIERCELIN commited on
Commit
da8972b
·
verified ·
1 Parent(s): 0b2a02e

Commented out the process to find the root cause

Browse files
Files changed (1) hide show
  1. app.py +14 -0
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: