@julien-c proposal for TF/PT compat in hf_buckets

2020-02-07 08:53:17 +01:00
parent d311f87bca
commit 961c69776f
2 changed files with 2 additions and 10 deletions
--- a/src/transformers/modeling_utils.py
+++ b/src/transformers/modeling_utils.py
@@ -421,11 +421,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin):
                )
                archive_file = pretrained_model_name_or_path + ".index"
            else:
-                archive_file = hf_bucket_url(pretrained_model_name_or_path, postfix=WEIGHTS_NAME)
-                if from_tf:
-                    raise EnvironmentError(
-                        "Loading a PyTorch model from a TF checkpoint is not supported when using a model identifier name."
-                    )
+                archive_file = hf_bucket_url(pretrained_model_name_or_path, postfix=(TF2_WEIGHTS_NAME if from_tf else WEIGHTS_NAME))

            # redirect to the cache, if necessary
            try: