【Sync】Release/2.0.1 (#2745)

* 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

View File

@@ -28,12 +28,12 @@ class ModelRegistry:
_registry = {}
@classmethod
def register(cls, model_class, suffix=""):
def register(cls, model_class):
"""register model class"""
if issubclass(
model_class,
ModelForCasualLM) and model_class is not ModelForCasualLM:
cls._registry[f"{model_class.name()}{suffix}"] = model_class
cls._registry[model_class.name()] = model_class
return model_class
@classmethod