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: