Fix CompressedTensorsMxInt4MoE abstract method and relax GPQA baseline (#19726)
Co-authored-by: Alison Shao <alisonshao@Mac.attlocal.net>
This commit is contained in:
co-authored by
Alison Shao
parent
8dfb6e1684
commit
fe9d85d93c
+1
-1
@@ -287,7 +287,7 @@ class CompressedTensorsMxInt4MoE(CompressedTensorsMoEScheme):
|
|||||||
):
|
):
|
||||||
self.moe_runner_config = moe_runner_config
|
self.moe_runner_config = moe_runner_config
|
||||||
|
|
||||||
def apply(
|
def apply_weights(
|
||||||
self,
|
self,
|
||||||
layer: torch.nn.Module,
|
layer: torch.nn.Module,
|
||||||
dispatch_output: StandardDispatchOutput,
|
dispatch_output: StandardDispatchOutput,
|
||||||
|
|||||||
@@ -30,7 +30,7 @@ DP_ARGS = [
|
|||||||
|
|
||||||
# Accuracy thresholds
|
# Accuracy thresholds
|
||||||
GSM8K_BASELINE = 0.935
|
GSM8K_BASELINE = 0.935
|
||||||
GPQA_BASELINE = 0.835
|
GPQA_BASELINE = 0.83
|
||||||
|
|
||||||
|
|
||||||
class TestDeepseekV32(unittest.TestCase):
|
class TestDeepseekV32(unittest.TestCase):
|
||||||
|
|||||||
Reference in New Issue
Block a user