Fix llava with multi-image inputs (#12384)

intel-analytics · Nov 13, 2024 · 1158f91 · 1158f91
1 parent 2715247
commit 1158f91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/llm/src/ipex_llm/transformers/low_bit_linear.py b/python/llm/src/ipex_llm/transformers/low_bit_linear.py
@@ -727,7 +727,7 @@ def forward(self, x: torch.Tensor):
             # return empty tensor with output shape, x.dtype and x.device
             return torch.empty(new_shape, dtype=x.dtype, device=x.device)
 
-        x_2d = x.view(-1, x_shape[-1])
+        x_2d = x.contiguous().view(-1, x_shape[-1])
 
         if self.act_order:
             x_2d = x_2d[:, self.g_idx_map]