@@ -77,7 +77,7 @@ def set_qwen3_vl_common_configs(cfg: ConfigContainer) -> None:
7777 cfg .comm_overlap .overlap_grad_reduce = False
7878
7979
80- def qwen3_vl_235b_a22b_pretrain_mock_config_gb300 (
80+ def qwen3_vl_235b_a22b_pretrain_config_gb300 (
8181 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
8282) -> ConfigContainer :
8383 """GB300, baseline config."""
@@ -102,7 +102,7 @@ def qwen3_vl_235b_a22b_pretrain_mock_config_gb300(
102102 return cfg
103103
104104
105- def qwen3_vl_235b_a22b_pretrain_mock_config_gb200 (
105+ def qwen3_vl_235b_a22b_pretrain_config_gb200 (
106106 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
107107) -> ConfigContainer :
108108 """GB200, baseline config."""
@@ -127,7 +127,7 @@ def qwen3_vl_235b_a22b_pretrain_mock_config_gb200(
127127 return cfg
128128
129129
130- def qwen3_vl_235b_a22b_pretrain_mock_config_b200 (
130+ def qwen3_vl_235b_a22b_pretrain_config_b200 (
131131 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
132132) -> ConfigContainer :
133133 """B200, baseline config."""
@@ -157,7 +157,7 @@ def qwen3_vl_235b_a22b_pretrain_mock_config_b200(
157157 return cfg
158158
159159
160- def qwen3_vl_235b_a22b_pretrain_mock_config_h100 (
160+ def qwen3_vl_235b_a22b_pretrain_config_h100 (
161161 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
162162) -> ConfigContainer :
163163 """H100, baseline config."""
@@ -180,7 +180,7 @@ def qwen3_vl_235b_a22b_pretrain_mock_config_h100(
180180 return cfg
181181
182182
183- def qwen3_vl_30b_a3b_pretrain_mock_config_gb300 (
183+ def qwen3_vl_30b_a3b_pretrain_config_gb300 (
184184 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
185185) -> ConfigContainer :
186186 """GB300, baseline config."""
@@ -205,7 +205,7 @@ def qwen3_vl_30b_a3b_pretrain_mock_config_gb300(
205205 return cfg
206206
207207
208- def qwen3_vl_30b_a3b_pretrain_mock_config_gb200 (
208+ def qwen3_vl_30b_a3b_pretrain_config_gb200 (
209209 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
210210) -> ConfigContainer :
211211 """GB200, baseline config."""
@@ -230,7 +230,7 @@ def qwen3_vl_30b_a3b_pretrain_mock_config_gb200(
230230 return cfg
231231
232232
233- def qwen3_vl_30b_a3b_pretrain_mock_config_b200 (
233+ def qwen3_vl_30b_a3b_pretrain_config_b200 (
234234 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
235235) -> ConfigContainer :
236236 """B200, baseline config."""
@@ -255,7 +255,7 @@ def qwen3_vl_30b_a3b_pretrain_mock_config_b200(
255255 return cfg
256256
257257
258- def qwen3_vl_30b_a3b_pretrain_mock_config_h100 (
258+ def qwen3_vl_30b_a3b_pretrain_config_h100 (
259259 precision : str = "bf16" , mock : bool = True , config_variant : str = "v1"
260260) -> ConfigContainer :
261261 """H100, baseline config."""
0 commit comments