add error traceback info (#3419)
Some checks failed
Deploy GitHub Pages / deploy (push) Has been cancelled

* add error traceback info

* update error msg

* update code

---------

Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
This commit is contained in:
kevin
2025-08-19 19:32:04 +08:00
committed by GitHub
parent b047681c5d
commit 67298cf4c0
30 changed files with 166 additions and 83 deletions

View File

@@ -15,6 +15,7 @@
"""
import os
import traceback
def check_safetensors_model(model_dir: str):
@@ -45,5 +46,5 @@ def check_safetensors_model(model_dir: str):
sum(flags) == safetensors_num
), f"Number of safetensor files should be {len(model_files)}, but now it's {sum(flags)}"
except Exception as e:
raise Exception(f"Failed to check unified checkpoint, details: {e}.")
raise Exception(f"Failed to check unified checkpoint, details: {e}, {str(traceback.format_exc())}.")
return is_safetensors