fix logger init (#1598)

InternLM · May 16, 2024 · 8f79144 · 8f79144
1 parent 14b6c02
commit 8f79144
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/utils.py b/lmdeploy/utils.py
@@ -265,6 +265,7 @@ def _get_and_verify_max_len(
     max_model_len: Optional[int],
 ) -> int:
     """Get and verify the model's maximum length."""
+    logger = get_logger('lmdeploy')
     derived_max_model_len = float('inf')
     possible_keys = [
         # OPT
@@ -295,7 +296,6 @@ def _get_and_verify_max_len(
             return max_model_len
 
         default_max_len = 2048
-        logger = get_logger('lmdeploy')
         logger.warning(
             "The model's config.json does not contain any of the following "
             'keys to determine the original maximum length of the model: '