[CI] fix lint error (#13891)
This commit is contained in:
@@ -150,7 +150,11 @@ def safetensors_weights_iterator(
|
|||||||
device = "cpu" if to_cpu else str(get_local_torch_device())
|
device = "cpu" if to_cpu else str(get_local_torch_device())
|
||||||
|
|
||||||
# Validate files before loading
|
# Validate files before loading
|
||||||
corrupted_files = [st_file for st_file in hf_weights_files if not _validate_safetensors_file(st_file)]
|
corrupted_files = [
|
||||||
|
st_file
|
||||||
|
for st_file in hf_weights_files
|
||||||
|
if not _validate_safetensors_file(st_file)
|
||||||
|
]
|
||||||
|
|
||||||
if corrupted_files:
|
if corrupted_files:
|
||||||
# Delete corrupted files (both symlink and blob if applicable)
|
# Delete corrupted files (both symlink and blob if applicable)
|
||||||
|
|||||||
Reference in New Issue
Block a user