apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

[CI] fix test_model_cache (#4982)

Browse Source 
* ci

* update
This commit is contained in:
bukejiyu
2025-11-12 20:26:49 +08:00
committed by GitHub
parent a2d06118e1
commit f0189292df
6 changed files with 12 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/load_weight_utils.py
+1 -1
View File
@@ -398,7 +398,7 @@ def get_all_weights_file(model_path: str):
"""
model_path = Path(model_path)
use_safetensors = True
files_list = [file for file in model_path.glob("*.pdparams") if file.name != "scheduler.pdparams"]
files_list = [str(file) for file in model_path.glob("*.pdparams") if file.name != "scheduler.pdparams"]
if len(files_list) > 0:
key_name_list = []
use_safetensors = False