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

polish code with new pre-commit rule (#2923)

Browse Source
This commit is contained in:
Zero Rains
2025-07-19 23:19:27 +08:00
committed by GitHub
parent b8676d71a8
commit 25698d56d1
424 changed files with 14307 additions and 13518 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
fastdeploy/worker/utils.py
View File

@@ -13,13 +13,14 @@
# See the License for the specific language governing permissions and
# limitations under the License.
"""
import os
def check_safetensors_model(model_dir: str):
"""
model_dir : the directory of the model
Check whther the model is safetensors format
model_dir : the directory of the model
Check whther the model is safetensors format
"""
model_files = list()
all_files = os.listdir(model_dir)
@@ -35,8 +36,7 @@ def check_safetensors_model(model_dir: str):
return True
try:
# check all the file exists
safetensors_num = int(
model_files[0].strip(".safetensors").split("-")[-1])
safetensors_num = int(model_files[0].strip(".safetensors").split("-")[-1])
flags = [0] * safetensors_num
for x in model_files:
current_index = int(x.strip(".safetensors").split("-")[1])