Unverified Commit 279d8e24 authored by Ori Garin's avatar Ori Garin Committed by GitHub
Browse files

support local_files_only option for tf models (#5116)

parent 355954ff
...@@ -356,6 +356,7 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin): ...@@ -356,6 +356,7 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
resume_download = kwargs.pop("resume_download", False) resume_download = kwargs.pop("resume_download", False)
proxies = kwargs.pop("proxies", None) proxies = kwargs.pop("proxies", None)
output_loading_info = kwargs.pop("output_loading_info", False) output_loading_info = kwargs.pop("output_loading_info", False)
local_files_only = kwargs.pop("local_files_only", False)
use_cdn = kwargs.pop("use_cdn", True) use_cdn = kwargs.pop("use_cdn", True)
# Load config if we don't provide a configuration # Load config if we don't provide a configuration
...@@ -368,6 +369,8 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin): ...@@ -368,6 +369,8 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
return_unused_kwargs=True, return_unused_kwargs=True,
force_download=force_download, force_download=force_download,
resume_download=resume_download, resume_download=resume_download,
proxies=proxies,
local_files_only=local_files_only,
**kwargs, **kwargs,
) )
else: else:
...@@ -405,8 +408,9 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin): ...@@ -405,8 +408,9 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
archive_file, archive_file,
cache_dir=cache_dir, cache_dir=cache_dir,
force_download=force_download, force_download=force_download,
resume_download=resume_download,
proxies=proxies, proxies=proxies,
resume_download=resume_download,
local_files_only=local_files_only,
) )
if resolved_archive_file is None: if resolved_archive_file is None:
raise EnvironmentError raise EnvironmentError
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment