riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix local_files_only for TF (#6091)

Browse Source
This commit is contained in:
Lysandre Debut 2020-10-01 11:06:02 +02:00 committed by GitHub
parent 9e80f972fb
commit 85d2d8c920
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/file_utils.py
View File

@ -847,7 +847,7 @@ def get_from_cache(
else: else:
matching_files = [ matching_files = [
file file
for file in fnmatch.filter(os.listdir(cache_dir), filename + ".*") for file in fnmatch.filter(os.listdir(cache_dir), filename.split(".")[0] + ".*")
if not file.endswith(".json") and not file.endswith(".lock") if not file.endswith(".json") and not file.endswith(".lock")
] ]
if len(matching_files) > 0: if len(matching_files) > 0: