drop _tied_weights_keys

Signed-off-by: Mayank Mishra <mayank31398@gmail.com>

Author: mayank31398

lm_engine/hf_models/modeling_utils/position_embedding/rope.py

@@ -2,8 +2,6 @@
 # Copyright (c) 2025, Mayank Mishra
 # **************************************************
 
-"""Logic is copied from transformers.models.llama.modeling_utils with slight modifications"""
-
 from __future__ import annotations
 
 import math

lm_engine/utils/safetensors.py

@@ -11,7 +11,6 @@
 from huggingface_hub import split_torch_state_dict_into_shards
 from safetensors import safe_open
 from safetensors.torch import save_file
-from transformers.modeling_utils import SAFE_WEIGHTS_INDEX_NAME
 
 
 _DEBUG = False
@@ -117,5 +116,5 @@ def save_state_dict(state_dict: dict, save_path: str) -> None:
                 "weight_map": state_dict_split.tensor_to_filename,
             }
 
-            with open(os.path.join(save_path, SAFE_WEIGHTS_INDEX_NAME), "w") as f:
+            with open(os.path.join(save_path, "model.safetensors.index.json"), "w") as f:
                 f.write(json.dumps(index, indent=2))