Update checkpoint.py typo

Jack-Khuu · web-flow · commit 774ebb6c2527 · 2024-11-14T08:15:15.000-08:00
diff --git a/torchchat/distributed/checkpoint.py b/torchchat/distributed/checkpoint.py
@@ -96,7 +96,7 @@ def _load_checkpoints_from_storage(
         checkpoint_path,
         map_location=builder_args.device,
         mmap=True,
-        weight_only=False,
+        weights_only=False,
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -96,7 +96,7 @@ def _load_checkpoints_from_storage(`
`96`	`96`	`checkpoint_path,`
`97`	`97`	`map_location=builder_args.device,`
`98`	`98`	`mmap=True,`
`99`		`- weight_only=False,`
	`99`	`+ weights_only=False,`
`100`	`100`	`)`
`101`	`101`
`102`	`102`