【Fearture】support qwen2 some func (#2740)

* add rl qwen model support

* fix

* fix
This commit is contained in:
gaoziyuan
2025-07-08 12:03:04 +08:00
committed by GitHub
parent fefbd65cf8
commit 26d5d737dd
13 changed files with 438 additions and 174 deletions

View File

@@ -36,8 +36,7 @@ def _find_py_files(root_dir):
def auto_models_registry(dir_path,
register_path="fastdeploy.model_executor.models",
suffix=""):
register_path="fastdeploy.model_executor.models"):
"""
auto registry all models in this folder
"""
@@ -49,7 +48,7 @@ def auto_models_registry(dir_path,
if inspect.isclass(attr) and issubclass(
attr,
ModelForCasualLM) and attr is not ModelForCasualLM:
ModelRegistry.register(attr, suffix=suffix)
ModelRegistry.register(attr)
except ImportError:
raise ImportError(f"{module_file=} import error")