Allow downloading of revisions for models. (#303)

2022-09-01 13:52:30 +02:00
parent 1e5d91d577
commit d799084a9a
2 changed files with 2 additions and 0 deletions
@@ -152,6 +152,7 @@ class ConfigMixin:
                    use_auth_token=use_auth_token,
                    user_agent=user_agent,
                    subfolder=subfolder,
+                    revision=revision,
                )

            except RepositoryNotFoundError:
@@ -373,6 +373,7 @@ class ModelMixin(torch.nn.Module):
                    use_auth_token=use_auth_token,
                    user_agent=user_agent,
                    subfolder=subfolder,
+                    revision=revision,
                )

            except RepositoryNotFoundError: