Fix local_files_only for TF (#6091)

2020-10-01 11:06:02 +02:00
parent 9e80f972fb
commit 85d2d8c920
1 changed files with 1 additions and 1 deletions
--- 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: