Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow downloading of revisions for models. (#303)

This commit is contained in:
okalldal 2022-09-01 13:52:30 +02:00 committed by GitHub
parent 1e5d91d577
commit d799084a9a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/configuration_utils.py
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:

1
src/diffusers/modeling_utils.py
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: