diff --git a/monai/transforms/utility/dictionary.py b/monai/transforms/utility/dictionary.py
index 3ef8cd4959..15025ac961 100644
--- a/monai/transforms/utility/dictionary.py
+++ b/monai/transforms/utility/dictionary.py
@@ -514,7 +514,7 @@ def __init__(
         self,
         keys: KeysCollection,
         dtype: torch.dtype | None = None,
-        device: torch.device | None = None,
+        device: torch.device | str | None = None,
         wrap_sequence: bool = True,
         track_meta: bool | None = None,
         allow_missing_keys: bool = False,