LAP-DEV commited on
Commit
e7f9f55
·
verified ·
1 Parent(s): 347e0c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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, whisper_enable_offload=whisper_params["enable_offload"], diarization_enable_offload=diarization_params["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,