Skip to content

Commit db77006

Browse files
committed
make style
1 parent 3643246 commit db77006

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -532,10 +532,10 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
532532
[`~DiffusionPipeline.save_pretrained`].
533533
torch_dtype (`str` or `torch.dtype` or `dict[str, Union[str, torch.dtype]]`, *optional*):
534534
Override the default `torch.dtype` and load the model with another dtype. If "auto" is passed, the
535-
dtype is automatically derived from the model's weights.
536-
To load submodels with different dtype pass a `dict` (for example `{'transformer': torch.bfloat16, 'vae': torch.float16}`).
537-
Set the default dtype for unspecified components with `_` (for example `{'transformer': torch.bfloat16, '_': torch.float16}`).
538-
If a component is not specifed and no default is set, `torch.float32` is used.
535+
dtype is automatically derived from the model's weights. To load submodels with different dtype pass a
536+
`dict` (for example `{'transformer': torch.bfloat16, 'vae': torch.float16}`). Set the default dtype for
537+
unspecified components with `_` (for example `{'transformer': torch.bfloat16, '_': torch.float16}`). If
538+
a component is not specifed and no default is set, `torch.float32` is used.
539539
custom_pipeline (`str`, *optional*):
540540
541541
<Tip warning={true}>

0 commit comments

Comments
 (0)