Fix local_files_only for TF (#6091)

85d2d8c9 · Lysandre Debut · GitHub · 9e80f972 · 85d2d8c9
Unverified Commit 85d2d8c9 authored Oct 01, 2020 by Lysandre Debut Committed by GitHub Oct 01, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/file_utils.py src/transformers/file_utils.py +1 -1

No files found.
--- a/src/transformers/file_utils.py
+++ b/src/transformers/file_utils.py
@@ -847,7 +847,7 @@ def get_from_cache(
        else:
            matching_files = [
                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 len(matching_files) > 0: