修复blip推理报错 (#1838)

mindspore-lab · Nov 23, 2024 · 0a936bf · 0a936bf
1 parent 756344b
commit 0a936bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindnlp/transformers/models/blip/modeling_blip.py b/mindnlp/transformers/models/blip/modeling_blip.py
@@ -1679,7 +1679,7 @@ def generate(
         question_attention_mask = ops.ones(*question_embeds.shape[:-1], dtype=mindspore.int64)
 
         bos_ids = ops.full(
-            (question_embeds.shape[0], 1), fill_value=self.decoder_start_token_id
+            (question_embeds.shape[0], 1), fill_value=self.decoder_start_token_id, dtype=mindspore.int64
         )
 
         outputs = self.text_decoder.generate(