[ONNX] Don't download ONNX model by default

[patrickvonplaten]

What does this PR do?

This PR makes sure that ONNX weights are only ever by default downloaded by ONNX pipelines, but not any PyTorch pipelines.

The user is also given the option to have more fine-grained control via a use_onnx download argument.

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[echarlaix]

Should this be set by default for all the OnnxStableDiffusionXxxPipeline ?

[echarlaix]

LGTM, thanks for the fix!

[JacquiML]

Hey @sayakpaul , in from_pretrained func, when calling cls.download, should we as use_onnx as an input parameter?
current:

 if not os.path.isdir(pretrained_model_name_or_path):
            cached_folder = cls.download(
                pretrained_model_name_or_path,
                cache_dir=cache_dir,
                resume_download=resume_download,
                force_download=force_download,
                proxies=proxies,
                local_files_only=local_files_only,
                use_auth_token=use_auth_token,
                revision=revision,
                from_flax=from_flax,
                use_safetensors=use_safetensors,
                custom_pipeline=custom_pipeline,
                custom_revision=custom_revision,
                variant=variant,
                load_connected_pipeline=load_connected_pipeline,
                **kwargs,
            )

proposed (added use_onnx=use_onnx,):

 if not os.path.isdir(pretrained_model_name_or_path):
            cached_folder = cls.download(
                pretrained_model_name_or_path,
                cache_dir=cache_dir,
                resume_download=resume_download,
                force_download=force_download,
                proxies=proxies,
                local_files_only=local_files_only,
                use_auth_token=use_auth_token,
                revision=revision,
                from_flax=from_flax,
                use_safetensors=use_safetensors,
                use_onnx=use_onnx,
                custom_pipeline=custom_pipeline,
                custom_revision=custom_revision,
                variant=variant,
                load_connected_pipeline=load_connected_pipeline,
                **kwargs,
            )

[sayakpaul]

@patrickvonplaten ^.

[patrickvonplaten]

Hey @JacquiML,

Good point! Would you like to open a PR for this?

[JacquiML]

Hey @JacquiML,

Good point! Would you like to open a PR for this?

sure :)