From 06350c31c780d6ea485f506032aea6ff8809e38a Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Tue, 12 Nov 2024 17:10:12 +0100
Subject: Add some missing index-select metal kernels. (#2613)

* Add some missing index-select metal kernels.

* Make some matrix contiguous pre-matmul.
---
 candle-transformers/src/models/chinese_clip/mod.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'candle-transformers')

diff --git a/candle-transformers/src/models/chinese_clip/mod.rs b/candle-transformers/src/models/chinese_clip/mod.rs
index 88472f0b..0f6eedd0 100644
--- a/candle-transformers/src/models/chinese_clip/mod.rs
+++ b/candle-transformers/src/models/chinese_clip/mod.rs
@@ -171,7 +171,8 @@ impl ChineseClipModel {
     ) -> Result<Tensor> {
         let output = self
             .text_model
-            .forward(input_ids, token_type_ids, attention_mask)?;
+            .forward(input_ids, token_type_ids, attention_mask)?
+            .contiguous()?;
         self.text_projection.forward(&output)
     }
 
-- 
cgit v1.2.3