jhj0517 commited on
Commit
6a24efd
·
1 Parent(s): 4171011

Opt-in offload

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -1
modules/whisper/whisper_base.py CHANGED
@@ -128,7 +128,8 @@ class WhisperBase(ABC):
128
  origin_sample_rate = self.music_separator.audio_info.sample_rate
129
  audio = self.resample_audio(audio=audio, original_sample_rate=origin_sample_rate)
130
 
131
- self.music_separator.offload()
 
132
 
133
  if params.vad_filter:
134
  # Explicit value set for float('inf') from gr.Number()
 
128
  origin_sample_rate = self.music_separator.audio_info.sample_rate
129
  audio = self.resample_audio(audio=audio, original_sample_rate=origin_sample_rate)
130
 
131
+ if params.uvr_enable_offload:
132
+ self.music_separator.offload()
133
 
134
  if params.vad_filter:
135
  # Explicit value set for float('inf') from gr.Number()