Skip to content

Commit e5820d7

Browse files
committed
fix doc
1 parent eb913e2 commit e5820d7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -814,7 +814,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
814814
Mirror source to resolve accessibility issues if you're downloading a model in China. We do not
815815
guarantee the timeliness or safety of the source, and you should refer to the mirror site for more
816816
information.
817-
device_map (`Union[int, str, torch.Device]` or `Dict[str, Union[int, str, torch.device]]`, *optional*):
817+
device_map (`Union[int, str, torch.device]` or `Dict[str, Union[int, str, torch.device]]`, *optional*):
818818
A map that specifies where each submodule should go. It doesn't need to be defined for each
819819
parameter/buffer name; once a given module name is inside, every submodule of it will be sent to the
820820
same device. Defaults to `None`, meaning that the model will be loaded on CPU.

0 commit comments

Comments
 (0)