DawnC commited on
Commit
0d6afde
·
verified ·
1 Parent(s): c56db8d

Upload FlowFacade.py

Browse files
Files changed (1) hide show
  1. FlowFacade.py +3 -3
FlowFacade.py CHANGED
@@ -41,13 +41,13 @@ class FlowFacade:
41
 
42
  # Add overhead for first-time model loading (FP8 quantization + AOTI)
43
  if not self.video_engine.is_loaded:
44
- total_duration += 60 # ~60s for FP8 quantization and AOTI loading
45
 
46
  if enable_prompt_expansion:
47
  total_duration += 40
48
 
49
- # Optimized minimum: 90 seconds (FP8 + AOTI is much faster)
50
- return max(int(total_duration), 90)
51
 
52
  @spaces.GPU(duration=_calculate_gpu_duration)
53
  def generate_video_from_image(self, image: Image.Image, user_instruction: str,
 
41
 
42
  # Add overhead for first-time model loading (FP8 quantization + AOTI)
43
  if not self.video_engine.is_loaded:
44
+ total_duration += 120 # ~120s for FP8 quantization, AOTI loading, and LoRA fusion
45
 
46
  if enable_prompt_expansion:
47
  total_duration += 40
48
 
49
+ # Conservative minimum: 180 seconds for first run, subsequent runs will be faster
50
+ return max(int(total_duration), 180)
51
 
52
  @spaces.GPU(duration=_calculate_gpu_duration)
53
  def generate_video_from_image(self, image: Image.Image, user_instruction: str,