apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-16 13:41:30 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

【Sync】Release/2.0.1 (#2745)

Browse Source 
* add rl qwen model support

* fix

* fix
This commit is contained in:
gaoziyuan
2025-07-08 14:38:18 +08:00
committed by GitHub
parent ea787d8f62
commit 6851489425
13 changed files with 438 additions and 171 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fastdeploy/model_executor/models/__init__.py
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")