Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
23bc56a0
Unverified
Commit
23bc56a0
authored
Feb 06, 2025
by
xieofxie
Committed by
GitHub
Feb 05, 2025
Browse files
add provider_options in from_pretrained (#10719)
Co-authored-by:
hualxie
<
hualxie@microsoft.com
>
parent
5b1dcd15
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
0 deletions
+4
-0
src/diffusers/pipelines/pipeline_loading_utils.py
src/diffusers/pipelines/pipeline_loading_utils.py
+2
-0
src/diffusers/pipelines/pipeline_utils.py
src/diffusers/pipelines/pipeline_utils.py
+2
-0
No files found.
src/diffusers/pipelines/pipeline_loading_utils.py
View file @
23bc56a0
...
...
@@ -630,6 +630,7 @@ def load_sub_model(
cached_folder
:
Union
[
str
,
os
.
PathLike
],
use_safetensors
:
bool
,
dduf_entries
:
Optional
[
Dict
[
str
,
DDUFEntry
]],
provider_options
:
Any
,
):
"""Helper method to load the module `name` from `library_name` and `class_name`"""
...
...
@@ -676,6 +677,7 @@ def load_sub_model(
if
issubclass
(
class_obj
,
diffusers_module
.
OnnxRuntimeModel
):
loading_kwargs
[
"provider"
]
=
provider
loading_kwargs
[
"sess_options"
]
=
sess_options
loading_kwargs
[
"provider_options"
]
=
provider_options
is_diffusers_model
=
issubclass
(
class_obj
,
diffusers_module
.
ModelMixin
)
...
...
src/diffusers/pipelines/pipeline_utils.py
View file @
23bc56a0
...
...
@@ -677,6 +677,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
custom_revision
=
kwargs
.
pop
(
"custom_revision"
,
None
)
provider
=
kwargs
.
pop
(
"provider"
,
None
)
sess_options
=
kwargs
.
pop
(
"sess_options"
,
None
)
provider_options
=
kwargs
.
pop
(
"provider_options"
,
None
)
device_map
=
kwargs
.
pop
(
"device_map"
,
None
)
max_memory
=
kwargs
.
pop
(
"max_memory"
,
None
)
offload_folder
=
kwargs
.
pop
(
"offload_folder"
,
None
)
...
...
@@ -971,6 +972,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
cached_folder
=
cached_folder
,
use_safetensors
=
use_safetensors
,
dduf_entries
=
dduf_entries
,
provider_options
=
provider_options
,
)
logger
.
info
(
f
"Loaded
{
name
}
as
{
class_name
}
from `
{
name
}
` subfolder of
{
pretrained_model_name_or_path
}
."
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment