apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-04 16:22:57 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Sync] Update to latest code (#2679)

Browse Source 
* [Sync] Update to latest code

* Add new code files

* Add new code files

* update code

* Try to fix build.sh

* Try to fix build.sh

* Update code

* Update requirements.txt

* Update code

---------

Co-authored-by: Jiang-Jia-Jun <jiangjiajun@baidu.com>
This commit is contained in:
Jiang-Jia-Jun
2025-07-03 15:43:53 +08:00
committed by GitHub
parent d222248d00
commit 05c670e593
95 changed files with 9916 additions and 1312 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/model_executor/models/model_base.py
View File

@@ -28,15 +28,17 @@ class ModelRegistry:
_registry = {}
@classmethod
def register(cls, model_class):
def register(cls, model_class, suffix=""):
"""register model class"""
if issubclass(
model_class,
ModelForCasualLM) and model_class is not ModelForCasualLM:
cls._registry[model_class.name()] = model_class
cls._registry[f"{model_class.name()}{suffix}"] = model_class
return model_class
@classmethod
def get_class(cls, name):
"""get model class"""
if name not in cls._registry:
raise ValueError(f"Model '{name}' is not registered!")
return cls._registry[name]