From 85d2d8c9200f8beff77544240b1da4c71d26371d Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 1 Oct 2020 11:06:02 +0200 Subject: [PATCH] Fix local_files_only for TF (#6091) --- src/transformers/file_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index f3844cef79..351d74f91a 100644 --- 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: