Use logger.exception() instead of traceback.print_exc() for error messages

This commit is contained in:
oobabooga 2026-03-18 20:42:20 -07:00
parent eeb0e5700f
commit 779e7611ff
10 changed files with 23 additions and 38 deletions

View file

@ -546,10 +546,8 @@ def do_train(lora_name: str, always_override: bool, all_linear: bool, q_proj_en:
yield f"Failed to load {selected_model}."
return
except Exception:
exc = traceback.format_exc()
logger.error('Failed to reload the model.')
print(exc)
yield exc.replace('\n', '\n\n')
logger.exception('Failed to reload the model.')
yield traceback.format_exc().replace('\n', '\n\n')
return
# == Start prepping the model itself ==