rename

legitnull · legitnull · commit 19227d0194f9 · 2026-03-13T18:06:55.000+08:00
diff --git a/examples/qwen_gr00t/conf/train.yaml b/examples/qwen_gr00t/conf/train.yaml
@@ -3,7 +3,7 @@ defaults:
   - train: qwen_gr00t
 
 experiment:
-  exp_name: qwen_gr00t_train
+  exp_name: 0313_qwen_gr00t_train_refactor
   seed: 42
   save_steps: 10000
   load: null
diff --git a/examples/qwen_gr00t/conf/train/qwen_gr00t.yaml b/examples/qwen_gr00t/conf/train/qwen_gr00t.yaml
@@ -12,7 +12,7 @@ system:
     # Whether to save checkpoint
     save_checkpoint: true
     # Number of steps between checkpoints
-    save_freq: 1000
+    save_freq: 1
   # TODO(yupu): Support resuming from checkpoint
 
 model:
diff --git a/flagscale/models/vla/action_model/flow_matching.py b/flagscale/models/vla/action_model/flow_matching.py
diff --git a/flagscale/models/vla/qwen_gr00t/__init__.py b/flagscale/models/vla/qwen_gr00t/__init__.py
@@ -0,0 +1,4 @@
+from .configuration_qwen_gr00t import QwenGr00tConfig
+from .modeling_qwen_gr00t import QwenGr00t
+
+__all__ = ["QwenGr00tConfig", "QwenGr00t"]
diff --git a/flagscale/models/vla/qwen_gr00t/configuration_qwen_gr00t.py b/flagscale/models/vla/qwen_gr00t/configuration_qwen_gr00t.py
@@ -8,7 +8,7 @@
 from flagscale.models.utils.constants import ACTION
 from flagscale.models.vla.action_model.gr00t_action_header import GR00TActionHeadConfig
 from flagscale.models.vla.pretrained_config import PreTrainedConfig
-from flagscale.models.vla.vlm.qwen_vl import QwenVLConfig
+from flagscale.models.vla.vlm.qwenvl_backbone import QwenVLConfig
 
 if TYPE_CHECKING:
     from flagscale.train.train_config import TrainConfig
diff --git a/flagscale/models/vla/qwen_gr00t/modeling_qwen_gr00t.py b/flagscale/models/vla/qwen_gr00t/modeling_qwen_gr00t.py
@@ -19,14 +19,14 @@
 import torch
 from safetensors.torch import save_file
 
+from .configuration_qwen_gr00t import QwenGr00tConfig
 from flagscale.models.utils.constants import (
     ACTION,
     OBS_STATE,
     VLM_CONFIG_DIR,
     resolve_pretrained_dir,
 )
 from flagscale.models.vla.base_policy import SAFETENSORS_FILE, TrainablePolicy
-from flagscale.models.vla.configuration_qwen_gr00t import QwenGr00tConfig
 from flagscale.models.vla.registry import build_action_model, build_vlm
 from flagscale.models.vla.utils import get_vlm_config
 
diff --git a/flagscale/models/vla/vlm/__init__.py b/flagscale/models/vla/vlm/__init__.py
@@ -1,3 +1,3 @@
-from .qwen_vl import Qwen3VLBackbone, Qwen25VLBackbone, QwenVLBackbone
+from .qwenvl_backbone import Qwen3VLBackbone, Qwen25VLBackbone, QwenVLBackbone
 
 __all__ = ["QwenVLBackbone", "Qwen25VLBackbone", "Qwen3VLBackbone"]
diff --git a/flagscale/models/vla/vlm/qwenvl_backbone.py b/flagscale/models/vla/vlm/qwenvl_backbone.py
diff --git a/flagscale/serve/__init__.py b/flagscale/serve/__init__.py
@@ -1,4 +1,3 @@
 from .utils import load_args, task_config
-from .websocket_policy_server import ServablePolicy
 
-__all__ = ["task_config", "load_args", "ServablePolicy"]
+__all__ = ["task_config", "load_args"]
diff --git a/flagscale/train/train_qwen_gr00t.py b/flagscale/train/train_qwen_gr00t.py
@@ -445,7 +445,7 @@ def main(config: TrainConfig, seed: int):
 
     if config.data.dataset_type == "wds":
         from megatron.energon import get_train_dataset, get_loader, WorkerConfig
-        from flagscale.models.vla.configuration_qwen_gr00t import QwenGr00tConfig
+        from flagscale.models.vla.qwen_gr00t import QwenGr00tConfig
         from flagscale.models.vla.qwen_gr00t_task_encoder import TaskEncoder
 
         if not isinstance(policy_config, QwenGr00tConfig):
diff --git a/tests/unit_tests/models/vla/vlm/test_qwen_vl.py b/tests/unit_tests/models/vla/vlm/test_qwen_vl.py
@@ -5,24 +5,24 @@
 
 class TestQwenVLRegistration(unittest.TestCase):
     def test_qwen25_vl_registered(self):
-        from flagscale.models.vla.vlm import qwen_vl  # noqa: F401
+        from flagscale.models.vla.vlm import qwenvl_backbone  # noqa: F401
 
         self.assertIn("qwen2.5-vl", VLM_REGISTRY)
 
     def test_qwen3_vl_registered(self):
-        from flagscale.models.vla.vlm import qwen_vl  # noqa: F401
+        from flagscale.models.vla.vlm import qwenvl_backbone  # noqa: F401
 
         self.assertIn("qwen3-vl", VLM_REGISTRY)
 
     def test_qwen25_has_required_methods(self):
-        from flagscale.models.vla.vlm.qwen_vl import Qwen25VLBackbone
+        from flagscale.models.vla.vlm.qwenvl_backbone import Qwen25VLBackbone
 
         self.assertTrue(hasattr(Qwen25VLBackbone, "model_config"))
         self.assertTrue(hasattr(Qwen25VLBackbone, "prepare_input"))
         self.assertTrue(hasattr(Qwen25VLBackbone, "forward"))
 
     def test_qwen3_has_required_methods(self):
-        from flagscale.models.vla.vlm.qwen_vl import Qwen3VLBackbone
+        from flagscale.models.vla.vlm.qwenvl_backbone import Qwen3VLBackbone
 
         self.assertTrue(hasattr(Qwen3VLBackbone, "model_config"))
         self.assertTrue(hasattr(Qwen3VLBackbone, "prepare_input"))

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,3 @@`
`1`		`-from .qwen_vl import Qwen3VLBackbone, Qwen25VLBackbone, QwenVLBackbone`
	`1`	`+from .qwenvl_backbone import Qwen3VLBackbone, Qwen25VLBackbone, QwenVLBackbone`
`2`	`2`
`3`	`3`	`__all__ = ["QwenVLBackbone", "Qwen25VLBackbone", "Qwen3VLBackbone"]`