Skip to content

Commit

Permalink
fix: bug where the logger was not being used properly (#286)
Browse files Browse the repository at this point in the history
Signed-off-by: Hari <harikrishmenon@gmail.com>
  • Loading branch information
HarikrishnanBalagopal authored Aug 5, 2024
1 parent e0da345 commit 5853949
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tuning/sft_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -515,7 +515,7 @@ def main(**kwargs): # pylint: disable=unused-argument
exp_metadata,
)
except Exception as e: # pylint: disable=broad-except
logging.error(traceback.format_exc())
logger.error(traceback.format_exc())
write_termination_log(
f"Exception raised during training. This may be a problem with your input: {e}"
)
Expand Down

0 comments on commit 5853949

Please sign in to comment.