jhj0517 commited on
Commit
790409b
·
1 Parent(s): b724669

Add `uvr_enable_offload` component

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -138,6 +138,8 @@ class App:
138
  choices=self.whisper_inf.music_separator.available_models)
139
  nb_uvr_segment_size = gr.Number(label="Segment Size", value=uvr_params["segment_size"], precision=0)
140
  cb_uvr_save_file = gr.Checkbox(label="Save separated files to output", value=uvr_params["save_file"])
 
 
141
 
142
  with gr.Accordion("VAD", open=False):
143
  cb_vad_filter = gr.Checkbox(label="Enable Silero VAD Filter", value=vad_params["vad_filter"],
@@ -187,7 +189,7 @@ class App:
187
  language_detection_segments=nb_language_detection_segments,
188
  prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
189
  uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
190
- uvr_save_file=cb_uvr_save_file
191
  ),
192
  dd_file_format,
193
  cb_timestamp
 
138
  choices=self.whisper_inf.music_separator.available_models)
139
  nb_uvr_segment_size = gr.Number(label="Segment Size", value=uvr_params["segment_size"], precision=0)
140
  cb_uvr_save_file = gr.Checkbox(label="Save separated files to output", value=uvr_params["save_file"])
141
+ cb_uvr_enable_offload = gr.Checkbox(label="Offload UVR model after separation",
142
+ value=uvr_params["enable_offload"])
143
 
144
  with gr.Accordion("VAD", open=False):
145
  cb_vad_filter = gr.Checkbox(label="Enable Silero VAD Filter", value=vad_params["vad_filter"],
 
189
  language_detection_segments=nb_language_detection_segments,
190
  prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
191
  uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
192
+ uvr_save_file=cb_uvr_save_file, uvr_enable_offload=cb_uvr_enable_offload
193
  ),
194
  dd_file_format,
195
  cb_timestamp