修复blip推理报错 (#1838)

confused666 · web-flow · commit 0a936bfc5f2e · 2024-11-23T21:20:33.000+08:00
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(

Original file line number	Diff line number	Diff line change
`@@ -1679,7 +1679,7 @@ def generate(`
`1679`	`1679`	`question_attention_mask = ops.ones(*question_embeds.shape[:-1], dtype=mindspore.int64)`
`1680`	`1680`
`1681`	`1681`	`bos_ids = ops.full(`
`1682`		`- (question_embeds.shape[0], 1), fill_value=self.decoder_start_token_id`
	`1682`	`+ (question_embeds.shape[0], 1), fill_value=self.decoder_start_token_id, dtype=mindspore.int64`
`1683`	`1683`	`)`
`1684`	`1684`
`1685`	`1685`	`outputs = self.text_decoder.generate(`