@@ -158,9 +158,9 @@ def is_safetensors_compatible(filenames, variant=None, passed_components=None) -
158
158
continue
159
159
160
160
if extension == ".bin" :
161
- pt_filenames .append (filename )
161
+ pt_filenames .append (os . path . normpath ( filename ) )
162
162
elif extension == ".safetensors" :
163
- sf_filenames .add (filename )
163
+ sf_filenames .add (os . path . normpath ( filename ) )
164
164
165
165
for filename in pt_filenames :
166
166
# filename = 'foo/bar/baz.bam' -> path = 'foo/bar', filename = 'baz', extention = '.bam'
@@ -172,9 +172,8 @@ def is_safetensors_compatible(filenames, variant=None, passed_components=None) -
172
172
else :
173
173
filename = filename
174
174
175
- expected_sf_filename = os .path .join (path , filename )
175
+ expected_sf_filename = os .path .normpath ( os . path . join (path , filename ) )
176
176
expected_sf_filename = f"{ expected_sf_filename } .safetensors"
177
-
178
177
if expected_sf_filename not in sf_filenames :
179
178
logger .warning (f"{ expected_sf_filename } not found" )
180
179
return False
@@ -1774,7 +1773,7 @@ def download(cls, pretrained_model_name, **kwargs) -> Union[str, os.PathLike]:
1774
1773
)
1775
1774
):
1776
1775
raise EnvironmentError (
1777
- f"Could not found the necessary `safetensors` weights in { model_filenames } (variant={ variant } )"
1776
+ f"Could not find the necessary `safetensors` weights in { model_filenames } (variant={ variant } )"
1778
1777
)
1779
1778
if from_flax :
1780
1779
ignore_patterns = ["*.bin" , "*.safetensors" , "*.onnx" , "*.pb" ]
0 commit comments