Update processing_sailvl.py

processing_sailvl.py

@@ -128,10 +128,11 @@ class SAILVLProcessor(ProcessorMixin):
             raise ValueError("Invalid input text. Please provide a string, or a list of strings")
 
         query = text[0]
-        for num_patches in image_inputs['num_patches_list']:
-            image_tokens = '<img>' + '<IMG_CONTEXT>' * self.num_image_token * num_patches + '</img>'
-            query = query.replace('<image>', image_tokens, 1)
-        image_inputs.pop('num_patches_list')
+        if images is not None:
+            for num_patches in image_inputs['num_patches_list']:
+                image_tokens = '<img>' + '<IMG_CONTEXT>' * self.num_image_token * num_patches + '</img>'
+                query = query.replace('<image>', image_tokens, 1)
+            image_inputs.pop('num_patches_list')
 
         model_inputs = self.tokenizer(query, return_tensors='pt')
         input_ids = model_inputs['input_ids'].cuda()