ALWOAL commited on
Commit
46639f4
·
verified ·
1 Parent(s): 9923301

TEMP: traceback for native warmup NotImplementedError

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -206,7 +206,12 @@ def _encode_decode(audio_path: str, levels: tuple) -> dict:
206
  with torch.inference_mode(): # untimed warmup so the first row's RTF isn't inflated
207
  codec.sig_to_sig(sig)
208
  if NATIVE_LEVEL in focal_levels:
209
- codec_i8.sig_to_sig(sig)
 
 
 
 
 
210
  for level in focal_levels:
211
  model = codec_i8 if level == NATIVE_LEVEL else codec
212
  if level != NATIVE_LEVEL:
 
206
  with torch.inference_mode(): # untimed warmup so the first row's RTF isn't inflated
207
  codec.sig_to_sig(sig)
208
  if NATIVE_LEVEL in focal_levels:
209
+ try:
210
+ codec_i8.sig_to_sig(sig)
211
+ except NotImplementedError: # TEMP triage
212
+ import traceback
213
+
214
+ raise RuntimeError(traceback.format_exc()) from None
215
  for level in focal_levels:
216
  model = codec_i8 if level == NATIVE_LEVEL else codec
217
  if level != NATIVE_LEVEL: