Allow downloading of revisions for models. (#303)

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

            except RepositoryNotFoundError:
--- a/src/diffusers/modeling_utils.py
+++ b/src/diffusers/modeling_utils.py
@ -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: