From 56ba132fbded7afcac48966916e9c01b4a2f9c8d Mon Sep 17 00:00:00 2001 From: teith Date: Thu, 30 Oct 2025 14:13:37 +0000 Subject: [PATCH] compilation fix --- ovi/modules/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ovi/modules/model.py b/ovi/modules/model.py index f1c16a5..3f94ded 100644 --- a/ovi/modules/model.py +++ b/ovi/modules/model.py @@ -689,6 +689,7 @@ def set_rope_params(self): def set_gradient_checkpointing(self, enable: bool): self.gradient_checkpointing = enable + @torch.compiler.disable def prepare_transformer_block_kwargs( self, x, @@ -802,7 +803,8 @@ def prepare_transformer_block_kwargs( context_lens=context_lens) return x, e, kwargs - + + @torch.compiler.disable def post_transformer_block_out(self, x, grid_sizes, e): # head x = self.head(x, e)