From d799084a9a7d5196ae708da72b46dd8e84604194 Mon Sep 17 00:00:00 2001
From: okalldal <oscar@kalldal.se>
Date: Thu, 1 Sep 2022 13:52:30 +0200
Subject: [PATCH] Allow downloading of revisions for models. (#303)

---
 src/diffusers/configuration_utils.py | 1 +
 src/diffusers/modeling_utils.py      | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/diffusers/configuration_utils.py b/src/diffusers/configuration_utils.py
index 646bbf22..053ccd64 100644
--- 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:
diff --git a/src/diffusers/modeling_utils.py b/src/diffusers/modeling_utils.py
index a8489a7b..ec501e2a 100644
--- 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: