minor: update error message

DNXie · DNXie · commit e3a1061af283 · 2025-08-18T10:51:56.000-07:00
diff --git a/apps/sft/main.py b/apps/sft/main.py
@@ -313,7 +313,7 @@ def batch_to_device(batch: dict, device: torch.device) -> None:
             batch[k] = v.to(device)
         else:
             raise ValueError(
-                f"""To use batch_to_device, all elements in the batch must be a dict or Tensor.
+                f"""To use batch_to_device, all elements in the batch must be a dict, Tensor, or BlockMask with flexattention enabled.
 Got key "{k}" with value of type {type(v)}"""
             )
 

Original file line number	Diff line number	Diff line change
`@@ -313,7 +313,7 @@ def batch_to_device(batch: dict, device: torch.device) -> None:`
`313`	`313`	`batch[k] = v.to(device)`
`314`	`314`	`else:`
`315`	`315`	`raise ValueError(`
`316`		`- f"""To use batch_to_device, all elements in the batch must be a dict or Tensor.`
	`316`	`+ f"""To use batch_to_device, all elements in the batch must be a dict, Tensor, or BlockMask with flexattention enabled.`
`317`	`317`	`Got key "{k}" with value of type {type(v)}"""`
`318`	`318`	`)`
`319`	`319`