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

[plugin] Custom model_runner/model support (#3186)

Browse Source 
* support custom model&&model_runner

* fix merge

* add test && update doc

* fix codestyle

* fix unittest

* load model in rl
This commit is contained in:
lizhenyun01
2025-08-05 09:52:39 +08:00
committed by GitHub
parent 72ef5a9c93
commit fe540f6caa
15 changed files with 150 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -64,9 +64,9 @@ class ModelRegistry:
@classmethod
def get_supported_archs(cls):
assert len(cls._arch_to_model_cls) == len(
cls._arch_to_model_cls
), "model class / pretrained model registry num is not same"
assert len(cls._arch_to_model_cls) >= len(
cls._arch_to_pretrained_model_cls
), "model class num is more than pretrained model registry num"
return [key for key in cls._arch_to_model_cls.keys()]