Reduce excessively long logs caused by transformer version updates. (#25737)
Co-authored-by: Adarsh Shirawalmath <114558126+adarshxs@users.noreply.github.com>
This commit is contained in:
co-authored by
Adarsh Shirawalmath
parent
1fbee74fb6
commit
579fed2090
@@ -1,3 +1,4 @@
|
|||||||
|
import os
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from sglang.test.ascend.test_ascend_utils import (
|
from sglang.test.ascend.test_ascend_utils import (
|
||||||
@@ -65,6 +66,8 @@ class TestNpuHierarchicalCacheMla(CustomTestCase):
|
|||||||
"write_back",
|
"write_back",
|
||||||
],
|
],
|
||||||
]
|
]
|
||||||
|
|
||||||
|
os.environ["TRANSFORMERS_VERBOSITY"] = "error"
|
||||||
for common_arg in common_args:
|
for common_arg in common_args:
|
||||||
other_args = common_arg + (
|
other_args = common_arg + (
|
||||||
[
|
[
|
||||||
|
|||||||
+1
@@ -59,6 +59,7 @@ class TestAscendDeepEP(CustomTestCase):
|
|||||||
"HCCL_BUFFSIZE": "1000",
|
"HCCL_BUFFSIZE": "1000",
|
||||||
"SGLANG_DEEPEP_NUM_MAX_DISPATCH_TOKENS_PER_RANK": "32",
|
"SGLANG_DEEPEP_NUM_MAX_DISPATCH_TOKENS_PER_RANK": "32",
|
||||||
"SGLANG_NPU_USE_MLAPO": "1",
|
"SGLANG_NPU_USE_MLAPO": "1",
|
||||||
|
"TRANSFORMERS_VERBOSITY": "error",
|
||||||
}
|
}
|
||||||
os.environ.update(cls.extra_envs)
|
os.environ.update(cls.extra_envs)
|
||||||
|
|
||||||
|
|||||||
+1
@@ -53,6 +53,7 @@ class TestDeepEpDeepseekV32(GSM8KAscendMixin, TestMMLU, CustomTestCase):
|
|||||||
"SGLANG_NPU_USE_MLAPO": "0",
|
"SGLANG_NPU_USE_MLAPO": "0",
|
||||||
"SGLANG_NPU_USE_MULTI_STREAM": "1",
|
"SGLANG_NPU_USE_MULTI_STREAM": "1",
|
||||||
"TASK_QUEUE_ENABLE": "0",
|
"TASK_QUEUE_ENABLE": "0",
|
||||||
|
"TRANSFORMERS_VERBOSITY": "error",
|
||||||
}
|
}
|
||||||
|
|
||||||
accuracy = 0.95 # Test GSM8K accuracy ≥0.95
|
accuracy = 0.95 # Test GSM8K accuracy ≥0.95
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
import os
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from sglang.test.ascend.gsm8k_ascend_mixin import GSM8KAscendMixin
|
from sglang.test.ascend.gsm8k_ascend_mixin import GSM8KAscendMixin
|
||||||
@@ -29,6 +30,7 @@ class TestQWQ32BW8A8(GSM8KAscendMixin, CustomTestCase):
|
|||||||
"--quantization",
|
"--quantization",
|
||||||
"modelslim",
|
"modelslim",
|
||||||
]
|
]
|
||||||
|
os.environ["TRANSFORMERS_VERBOSITY"] = "error"
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
@@ -63,6 +63,9 @@ class TestMemoryConsumptionAscend(CustomTestCase):
|
|||||||
"--disable-radix-cache",
|
"--disable-radix-cache",
|
||||||
"--disable-cuda-graph",
|
"--disable-cuda-graph",
|
||||||
],
|
],
|
||||||
|
env={
|
||||||
|
"TRANSFORMERS_VERBOSITY": "error",
|
||||||
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
### Calculate initial used memory
|
### Calculate initial used memory
|
||||||
|
|||||||
Reference in New Issue
Block a user