Revert "[Model(PaddleOCR-VL)] Disable preserve_external_rng_state in recomp…"

DrRyanHuang · web-flow · commit 8ac4aee53959 · 2026-03-20T16:36:50.000+08:00
This reverts commit a4a31fb.
diff --git a/paddleformers/transformers/paddleocr_vl/modeling.py b/paddleformers/transformers/paddleocr_vl/modeling.py
@@ -523,7 +523,6 @@ def recompute_training(
             cu_seqlens=cu_seqlens,
             attn_mask_startend_row_indices=attn_mask_startend_row_indices,
             rope_emb=rope_emb,
-            preserve_external_rng_state=False,
         )
         return hidden_states
 
@@ -1556,7 +1555,6 @@ def recompute_training(
             output_attentions,
             past_key_values,
             use_cache,
-            preserve_external_rng_state=False,
         )
         return hidden_states
 

Original file line number	Diff line number	Diff line change
`@@ -523,7 +523,6 @@ def recompute_training(`
`523`	`523`	`cu_seqlens=cu_seqlens,`
`524`	`524`	`attn_mask_startend_row_indices=attn_mask_startend_row_indices,`
`525`	`525`	`rope_emb=rope_emb,`
`526`		`- preserve_external_rng_state=False,`
`527`	`526`	`)`
`528`	`527`	`return hidden_states`
`529`	`528`
`@@ -1556,7 +1555,6 @@ def recompute_training(`
`1556`	`1555`	`output_attentions,`
`1557`	`1556`	`past_key_values,`
`1558`	`1557`	`use_cache,`
`1559`		`- preserve_external_rng_state=False,`
`1560`	`1558`	`)`
`1561`	`1559`	`return hidden_states`
`1562`	`1560`