Skip to content

Commit b8cf4ea

Browse files
fix: use cpu device for dcp to torch converted checkpoints
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 95cead4 commit b8cf4ea

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/modalities/checkpointing/convert_dcp_to_torch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def convert_config_file(dcp_checkpoint_dir: str, output_dir: str, model_key: str
6262
"component_key": "checkpoint_loading",
6363
"variant_key": "torch",
6464
"config": {
65-
"device": 0,
65+
"device": "cpu",
6666
"precision": "BF16", # FIXME Should this be configurable?
6767
},
6868
},

0 commit comments

Comments
 (0)