Skip to content

GitLab

"apps/life_sci/python/setup.py" did not exist on "e4948c5cbce13a054d639c5b4f0be20ba166d68e"
Unverified Commit d799084a authored by okalldal's avatar okalldal Committed by GitHub
Browse files

Allow downloading of revisions for models. (#303)

parent 1e5d91d5
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/configuration_utils.py
View file @ d799084a
...@@ -152,6 +152,7 @@ class ConfigMixin: ...@@ -152,6 +152,7 @@ class ConfigMixin:
use_auth_token=use_auth_token, use_auth_token=use_auth_token,
user_agent=user_agent, user_agent=user_agent,
subfolder=subfolder, subfolder=subfolder,
revision=revision,
) )
except RepositoryNotFoundError: except RepositoryNotFoundError:
......
src/diffusers/modeling_utils.py
View file @ d799084a
...@@ -373,6 +373,7 @@ class ModelMixin(torch.nn.Module): ...@@ -373,6 +373,7 @@ class ModelMixin(torch.nn.Module):
use_auth_token=use_auth_token, use_auth_token=use_auth_token,
user_agent=user_agent, user_agent=user_agent,
subfolder=subfolder, subfolder=subfolder,
revision=revision,
) )
except RepositoryNotFoundError: except RepositoryNotFoundError:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment