[Fix] configuration_internlm.py -> configuration_internlm2.py (#1129)

InternLM · Feb 6, 2024 · 81665a6 · 81665a6
1 parent c332efa
commit 81665a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lmdeploy/lite/apis/smooth_quant.py b/lmdeploy/lite/apis/smooth_quant.py
@@ -49,7 +49,7 @@
         'AutoModelForCausalLM': 'modeling_internlm.InternLMForCausalLM'
     },
     'InternLM2ForCausalLM': {
-        'AutoConfig': 'configuration_internlm.InternLMConfig',
+        'AutoConfig': 'configuration_internlm2.InternLMConfig',
         'AutoModelForCausalLM': 'modeling_internlm2.InternLM2ForCausalLM',
         'AutoModel': 'modeling_internlm2.InternLM2ForCausalLM'
     },

diff --git a/lmdeploy/pytorch/modeling/modeling_internlm2.py b/lmdeploy/pytorch/modeling/modeling_internlm2.py
@@ -45,7 +45,7 @@
 except:  # noqa # pylint: disable=bare-except
     BaseStreamer = None
 
-from .configuration_internlm import InternLMConfig as InternLM2Config
+from .configuration_internlm2 import InternLM2Config
 
 logger = get_logger('lmdeploy')