Fix haotian-liu#1050

tousika · Feb 7, 2024 · b42a13d · b42a13d
1 parent 49540fe
commit b42a13d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llava/model/llava_arch.py b/llava/model/llava_arch.py
@@ -192,7 +192,7 @@ def prepare_inputs_labels_for_multimodal(
                         if 'unpad' in mm_patch_merge_type:
                             image_feature = torch.cat((
                                 image_feature,
-                                self.model.image_newline[None]
+                                self.model.image_newline[None].to(image_feature.device)
                             ), dim=0)
                     new_image_features.append(image_feature)
                 image_features = new_image_features