Skip to content

Commit 81665a6

Browse files
authored
[Fix] configuration_internlm.py -> configuration_internlm2.py (#1129)
1 parent c332efa commit 81665a6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lmdeploy/lite/apis/smooth_quant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
'AutoModelForCausalLM': 'modeling_internlm.InternLMForCausalLM'
5050
},
5151
'InternLM2ForCausalLM': {
52-
'AutoConfig': 'configuration_internlm.InternLMConfig',
52+
'AutoConfig': 'configuration_internlm2.InternLMConfig',
5353
'AutoModelForCausalLM': 'modeling_internlm2.InternLM2ForCausalLM',
5454
'AutoModel': 'modeling_internlm2.InternLM2ForCausalLM'
5555
},

lmdeploy/pytorch/modeling/modeling_internlm2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
except: # noqa # pylint: disable=bare-except
4646
BaseStreamer = None
4747

48-
from .configuration_internlm import InternLMConfig as InternLM2Config
48+
from .configuration_internlm2 import InternLM2Config
4949

5050
logger = get_logger('lmdeploy')
5151

0 commit comments

Comments
 (0)