Skip to content

Commit

Permalink
nit fix
Browse files Browse the repository at this point in the history
  • Loading branch information
medivh-xp committed May 19, 2023
1 parent dd032d5 commit 2c13398
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/distributed/fsdp/_init_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -297,7 +297,7 @@ def _init_device_handle(
else:
if param.device.type != determined_device.type:
raise RuntimeError(
f"FSDP not supports modules on different device type "
f"FSDP does not support modules with different device types "
f"but got params on {determined_device.type} and {param.device.type}"
)
determined_device = determined_device or torch.device(
Expand Down

0 comments on commit 2c13398

Please sign in to comment.