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

[v1 loader]code style (#4204)

Browse Source 
* code style

* update
This commit is contained in:
bukejiyu
2025-09-23 19:36:00 +08:00
committed by GitHub
parent 1a6283424e
commit 62d1c48363
18 changed files with 33 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/model_executor/model_loader/default_loader_v1.py
View File

@@ -22,7 +22,7 @@ from fastdeploy.config import FDConfig, LoadConfig, ModelConfig
from fastdeploy.model_executor.load_weight_utils import (
get_weight_iterator,
is_weight_cache_enabled,
load_weights_form_cache,
load_weights_from_cache,
measure_time,
save_model,
)
@@ -52,7 +52,7 @@ class DefaultModelLoaderV1(BaseModelLoader):
def load_weights(self, model, fd_config: FDConfig, enable_cache: bool = False) -> None:
weights_iterator = get_weight_iterator(fd_config.model_config.model)
if enable_cache:
load_weights_form_cache(model, weights_iterator)
load_weights_from_cache(model, weights_iterator)
else:
model.load_weights(weights_iterator)