diff --git a/huggingface/internlm-xcomposer-4bit/modeling_InternLM_XComposer.py b/huggingface/internlm-xcomposer-4bit/modeling_InternLM_XComposer.py
index 4a43e42..6789726 100644
--- a/huggingface/internlm-xcomposer-4bit/modeling_InternLM_XComposer.py
+++ b/huggingface/internlm-xcomposer-4bit/modeling_InternLM_XComposer.py
@@ -378,7 +378,7 @@ def forward(self,
                 **kwargs):
 
         samples = kwargs.get('samples')
-        has_img = 'images' in samples.keys()
+        has_img = 'image' in samples.keys()
 
         ### encode text
         text = self.align_text(samples, has_img=has_img)
diff --git a/huggingface/internlm-xcomposer-vl/modeling_InternLM_XComposer.py b/huggingface/internlm-xcomposer-vl/modeling_InternLM_XComposer.py
index 4a43e42..6789726 100644
--- a/huggingface/internlm-xcomposer-vl/modeling_InternLM_XComposer.py
+++ b/huggingface/internlm-xcomposer-vl/modeling_InternLM_XComposer.py
@@ -378,7 +378,7 @@ def forward(self,
                 **kwargs):
 
         samples = kwargs.get('samples')
-        has_img = 'images' in samples.keys()
+        has_img = 'image' in samples.keys()
 
         ### encode text
         text = self.align_text(samples, has_img=has_img)
diff --git a/huggingface/internlm-xcomposer/modeling_InternLM_XComposer.py b/huggingface/internlm-xcomposer/modeling_InternLM_XComposer.py
index 4a43e42..6789726 100644
--- a/huggingface/internlm-xcomposer/modeling_InternLM_XComposer.py
+++ b/huggingface/internlm-xcomposer/modeling_InternLM_XComposer.py
@@ -378,7 +378,7 @@ def forward(self,
                 **kwargs):
 
         samples = kwargs.get('samples')
-        has_img = 'images' in samples.keys()
+        has_img = 'image' in samples.keys()
 
         ### encode text
         text = self.align_text(samples, has_img=has_img)