fix: logger is unbound error (foundation-model-stack#308)

* fix: logger is unbound error Signed-off-by: Harikrishnan Balagopal <[email protected]> * fixup! fix: logger is unbound error Signed-off-by: Harikrishnan Balagopal <[email protected]> --------- Signed-off-by: Harikrishnan Balagopal <[email protected]> Co-authored-by: Sukriti Sharma <[email protected]>
dushyantbehl · Aug 21, 2024 · daca551 · daca551
1 parent 822311c
commit daca551
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
@@ -510,6 +510,7 @@ def parse_arguments(parser, json_config=None):
 
 def main(**kwargs):  # pylint: disable=unused-argument
     parser = get_parser()
+    logger = logging.getLogger()
     job_config = get_json_config()
     # accept arguments via command-line or JSON
     try: