1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

add provider_options in from_pretrained (#10719)

Co-authored-by: hualxie <hualxie@microsoft.com>
This commit is contained in:
xieofxie
2025-02-06 03:41:41 +08:00
committed by GitHub
parent 5b1dcd1584
commit 23bc56a02d
2 changed files with 4 additions and 0 deletions

View File

@@ -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)

View File

@@ -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}."