refactor: remove unstable Option<&T> branch by returning owned Tensor

2025-12-11 10:50:35 +08:00
parent 0dc2840b0e
commit 42c95da0e3
16 changed files with 64 additions and 60 deletions
@@ -376,11 +376,11 @@ impl Ernie4_5Model {
            self.rope_scaling.mrope_section.clone(),
        )?;
        let mut xs = inputs_embeds.clone();
-        let attention_mask: Option<&Tensor> = {
+        let attention_mask: Option<Tensor> = {
            if seq_len <= 1 {
                None
            } else {
-                Some(&prepare_causal_attention_mask(
+                Some(prepare_causal_attention_mask(
                    b_size,
                    seq_len,
                    0,
@@ -389,7 +389,7 @@ impl Ernie4_5Model {
            }
        };
        for layer in self.layers.iter_mut() {
-            xs = layer.forward(&xs, &cos, &sin, attention_mask)?;
+            xs = layer.forward(&xs, &cos, &sin, attention_mask.as_ref())?;
        }
        let xs = xs.apply(&self.norm)?;
        Ok(xs)
@@ -564,7 +564,7 @@ impl PaddleOCRVLModel {
                        }
                    }
                    Err(e) => {
-                        println!("get vision_indices err: {}", e);
+                        println!("get vision_indices err: {e}");
                    }
                };