rahul7star commited on
Commit
79f7542
·
verified ·
1 Parent(s): 3417358

Update app_gradio_lora.py

Browse files
Files changed (1) hide show
  1. app_gradio_lora.py +14 -14
app_gradio_lora.py CHANGED
@@ -110,20 +110,20 @@ def load_loras_from_ui(selected_labels, weights, custom_repo, custom_file, custo
110
  pipe.fuse_lora()
111
  print(f"✅ Fused LoRAs: {lora_adapters}")
112
 
113
- def get_duration(
114
- prompt,
115
- nag_negative_prompt, nag_scale,
116
- height, width, duration_seconds,
117
- steps,
118
- seed, randomize_seed,
119
- compare,
120
- ):
121
- duration = int(duration_seconds) * int(steps) * 2.25 + 5
122
- if compare:
123
- duration *= 2
124
- return duration
125
-
126
- @spaces.GPU(duration=get_duration)
127
  def generate_video(prompt, nag_negative_prompt, nag_scale,
128
  height=DEFAULT_H_SLIDER_VALUE, width=DEFAULT_W_SLIDER_VALUE, duration_seconds=DEFAULT_DURATION_SECONDS,
129
  steps=DEFAULT_STEPS, seed=DEFAULT_SEED, randomize_seed=False, compare=True):
 
110
  pipe.fuse_lora()
111
  print(f"✅ Fused LoRAs: {lora_adapters}")
112
 
113
+ # def get_duration(
114
+ # prompt,
115
+ # nag_negative_prompt, nag_scale,
116
+ # height, width, duration_seconds,
117
+ # steps,
118
+ # seed, randomize_seed,
119
+ # compare,
120
+ # ):
121
+ # duration = int(duration_seconds) * int(steps) * 2.25 + 5
122
+ # if compare:
123
+ # duration *= 2
124
+ # return duration
125
+
126
+ @spaces.GPU(duration=200)
127
  def generate_video(prompt, nag_negative_prompt, nag_scale,
128
  height=DEFAULT_H_SLIDER_VALUE, width=DEFAULT_W_SLIDER_VALUE, duration_seconds=DEFAULT_DURATION_SECONDS,
129
  steps=DEFAULT_STEPS, seed=DEFAULT_SEED, randomize_seed=False, compare=True):