diff --git a/onediff_comfy_nodes/modules/oneflow/onediff_controlnet.py b/onediff_comfy_nodes/modules/oneflow/onediff_controlnet.py
index 25bfb4b36..1812e718b 100644
--- a/onediff_comfy_nodes/modules/oneflow/onediff_controlnet.py
+++ b/onediff_comfy_nodes/modules/oneflow/onediff_controlnet.py
@@ -40,7 +40,6 @@ def from_controllora(
         c = cls(
             controlnet.control_weights,
             global_average_pooling=controlnet.global_average_pooling,
-            device=controlnet.device,
         )
         controlnet.copy_to(c)
         c._oneflow_model = None
@@ -117,7 +116,6 @@ def copy(self):
         c = OneDiffControlLora(
             self.control_weights,
             global_average_pooling=self.global_average_pooling,
-            device=self.device,
         )
         self.copy_to(c)
         c._oneflow_model = self._oneflow_model