Allow downloading of revisions for models. (#303)

d799084a · okalldal · GitHub · 1e5d91d5 · d799084a · d799084a
Unverified Commit d799084a authored Sep 01, 2022 by okalldal Committed by GitHub Sep 01, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/diffusers/configuration_utils.py src/diffusers/configuration_utils.py +1 -0

src/diffusers/modeling_utils.py src/diffusers/modeling_utils.py +1 -0

No files found.
--- 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: