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

TEMP: traceback for native NotImplementedError

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -214,7 +214,12 @@ def _encode_decode(audio_path: str, levels: tuple) -> dict:
214
  torch.cuda.synchronize()
215
  start = time.perf_counter()
216
  with torch.inference_mode():
217
- rec = model.sig_to_sig(sig).squeeze(0).float().cpu().numpy()
 
 
 
 
 
218
  torch.cuda.synchronize()
219
  rtf = (time.perf_counter() - start) / seconds
220
  rows.append(core.evaluate(ref, rec, sr, rtf, SIZES[level], level))
 
214
  torch.cuda.synchronize()
215
  start = time.perf_counter()
216
  with torch.inference_mode():
217
+ try:
218
+ rec = model.sig_to_sig(sig).squeeze(0).float().cpu().numpy()
219
+ except NotImplementedError: # TEMP triage: surface the bare error's traceback
220
+ import traceback
221
+
222
+ raise RuntimeError(traceback.format_exc()) from None
223
  torch.cuda.synchronize()
224
  rtf = (time.perf_counter() - start) / seconds
225
  rows.append(core.evaluate(ref, rec, sr, rtf, SIZES[level], level))