tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: kwargs passsed

Browse Source
This commit is contained in:
NanoCode012
2025-12-22 18:13:28 +07:00
parent 001f4205f3
commit 9249b49a09
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/axolotl/monkeypatch/models/kimi_linear/patch_kimi_linear.py
View File

@@ -39,7 +39,7 @@ def patch_kimi_model():
# Store original function # Store original function
original_get_class_in_module = get_class_in_module original_get_class_in_module = get_class_in_module
def patched_get_class_in_module(class_name, module_path): def patched_get_class_in_module(class_name, module_path, **kwargs):
"""Patched version that returns our local modules instead of remote ones.""" """Patched version that returns our local modules instead of remote ones."""
# Check if this is a Kimi model module # Check if this is a Kimi model module
if "modeling_kimi" in module_path: if "modeling_kimi" in module_path:
@@ -68,8 +68,8 @@ def patch_kimi_model():
spec.loader.exec_module(module) spec.loader.exec_module(module)
return getattr(module, class_name) return getattr(module, class_name)
# For all other modules, use the original function # For all other modules, use the original function with all kwargs
return original_get_class_in_module(class_name, module_path) return original_get_class_in_module(class_name, module_path, **kwargs)
# Apply the monkey patch # Apply the monkey patch
import transformers.dynamic_module_utils import transformers.dynamic_module_utils