tie weights

modular_isaac.py

@@ -1196,6 +1196,8 @@ class IsaacRotaryEmbedding(nn.Module):
 
 
 class IsaacModel(Qwen3Model):
+    _tied_weights_keys = {}
+
     def __init__(self, config: IsaacConfig):
         super().__init__(config)
         text_cfg = getattr(config, "get_text_config", lambda: config)()
@@ -1379,6 +1381,7 @@ class IsaacForConditionalGeneration(Qwen3ForCausalLM, GenerationMixin):
     """Isaac multimodal model for conditional generation."""
 
     config_class = IsaacConfig
+    _tied_weights_keys = {"lm_head.weight": "model.embed_tokens.weight"}
 
     def __init__(self, config: IsaacConfig):
         Qwen3PreTrainedModel.__init__(self, config)