diff --git a/nerfactory/utils/misc.py b/nerfactory/utils/misc.py
index 297c5d536..a30a78992 100644
--- a/nerfactory/utils/misc.py
+++ b/nerfactory/utils/misc.py
@@ -29,8 +29,6 @@ def get_dict_to_torch(stuff: Any, device: Union[torch.device, str] = "cpu", excl
 
     Args:
         stuff: things to convert to torch
-        device: machine to put the "stuff" on
-        exclude: list of keys to skip over transferring to device
     """
     if isinstance(stuff, dict):
         for k, v in stuff.items():