Update app.py
Browse files
app.py
CHANGED
@@ -257,6 +257,9 @@ class App:
|
|
257 |
|
258 |
#dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
259 |
|
|
|
|
|
|
|
260 |
return (
|
261 |
WhisperParameters(
|
262 |
model_size=dd_model, lang=dd_lang, is_translate=cb_translate, beam_size=nb_beam_size,
|
@@ -278,7 +281,8 @@ class App:
|
|
278 |
language_detection_segments=nb_language_detection_segments,
|
279 |
prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
|
280 |
uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
|
281 |
-
uvr_save_file=cb_uvr_save_file, uvr_enable_offload=cb_uvr_enable_offload,
|
|
|
282 |
),
|
283 |
dd_file_format,
|
284 |
cb_timestamp,
|
|
|
257 |
|
258 |
#dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
259 |
|
260 |
+
bool_whisper_enable_offload = whisper_params["enable_offload"]
|
261 |
+
bool_diarization_enable_offload = diarization_params["enable_offload"]
|
262 |
+
|
263 |
return (
|
264 |
WhisperParameters(
|
265 |
model_size=dd_model, lang=dd_lang, is_translate=cb_translate, beam_size=nb_beam_size,
|
|
|
281 |
language_detection_segments=nb_language_detection_segments,
|
282 |
prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
|
283 |
uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
|
284 |
+
uvr_save_file=cb_uvr_save_file, uvr_enable_offload=cb_uvr_enable_offload,
|
285 |
+
whisper_enable_offload=bool_whisper_enable_offload, diarization_enable_offload=bool_diarization_enable_offload
|
286 |
),
|
287 |
dd_file_format,
|
288 |
cb_timestamp,
|