Unverified Commit 0fc8fa75 authored by Simon Mo's avatar Simon Mo Committed by GitHub
Browse files

fix: gptq marlin weight loading failure (#23066)

parent 21e39436
...@@ -56,7 +56,7 @@ def get_moe_quant_method( ...@@ -56,7 +56,7 @@ def get_moe_quant_method(
# Dynamic per module/layer rules may override base config # Dynamic per module/layer rules may override base config
override_config(cloned_config, prefix=prefix) override_config(cloned_config, prefix=prefix)
return moe_method_cls(cloned_config) return moe_method_cls(cloned_config, layer.moe_config)
return None return None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment