revert some change in decompose_util

Author: JewelRoam

graph_net/torch/decompose_util.py

@@ -262,8 +262,6 @@ def get_args_node(arg):
                     str,
                     type(...),
                     type(None),
-                    torch.device,
-                    torch.dtype,
                 ),
             ), f"{type(arg)=}"