ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-04 09:13:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[v1] fix kernel moe patch (#9867)

Browse Source
This commit is contained in:
jiaqiw09
2026-01-17 09:24:54 +08:00
committed by GitHub
parent 9829ae0a77
commit 44eadbda1c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/v1/plugins/model_plugins/kernels/ops/mlp/npu_fused_moe.py
View File

@@ -324,7 +324,7 @@ class NpuFusedMoEKernel(BaseKernel):
if not cls.check_deps(): if not cls.check_deps():
raise RuntimeError("torch_npu is not available but NpuMoEFusedMoEKernel was called.") raise RuntimeError("torch_npu is not available but NpuMoEFusedMoEKernel was called.")
archs = getattr(model.config, "architectures", []) archs = getattr(model.config, "architectures", None) or []
target_moe_mapping = None target_moe_mapping = None
for arch in archs: for arch in archs:
if arch in kernel_moe_mapping: if arch in kernel_moe_mapping: