kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow downloading of revisions for models. (#303)

Browse Source
This commit is contained in:
okalldal
2022-09-01 13:52:30 +02:00
committed by GitHub
parent 1e5d91d577
commit d799084a9a
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/configuration_utils.py
+1
View File
@@ -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
+1
View File
@@ -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: