fix AttributeError: 'MoonViT3dEncoder' object has no attribute 'use_deterministic_attn'

modeling_kimi_k25.py

@@ -562,7 +562,8 @@ class MoonViT3dEncoder(nn.Module):
                  hidden_dim: int,
                  num_layers: int,
                  block_cfg: dict,
-                 video_attn_type: str = 'spatial_temporal') -> None:
+                 video_attn_type: str = 'spatial_temporal',
+                 use_deterministic_attn: bool = False) -> None:
         super().__init__()
 
         assert video_attn_type == 'spatial_temporal', f'video_attn_type must be "spatial_temporal", got {video_attn_type}'
@@ -572,7 +573,7 @@ class MoonViT3dEncoder(nn.Module):
         self.blocks = nn.ModuleList([
             MoonViTEncoderLayer(
                 **block_cfg,
-                use_deterministic_attn=self.use_deterministic_attn)
+                use_deterministic_attn=use_deterministic_attn)
             for _ in range(num_layers)
         ])
         self.final_layernorm = nn.LayerNorm(hidden_dim)