diff --git a/DiT/loader.py b/DiT/loader.py
index 83db9b7..14c9f11 100644
--- a/DiT/loader.py
+++ b/DiT/loader.py
@@ -56,6 +56,5 @@ def load_dit(model_path, model_conf):
 		model,
 		load_device = load_device,
 		offload_device = offload_device,
-		current_device = "cpu",
 	)
 	return model_patcher
diff --git a/HunYuanDiT/loader.py b/HunYuanDiT/loader.py
index 8b89608..23be529 100644
--- a/HunYuanDiT/loader.py
+++ b/HunYuanDiT/loader.py
@@ -76,6 +76,5 @@ def load_hydit(model_path, model_conf):
 		model,
 		load_device = load_device,
 		offload_device = offload_device,
-		current_device = "cpu",
 	)
 	return model_patcher
diff --git a/HunYuanDiT/tenc.py b/HunYuanDiT/tenc.py
index 7f1633b..a2d7107 100644
--- a/HunYuanDiT/tenc.py
+++ b/HunYuanDiT/tenc.py
@@ -127,7 +127,6 @@ class EXM_HyDiT_Tenc_Temp:
 			self.cond_stage_model,
 			load_device    = self.load_device,
 			offload_device = self.offload_device,
-			current_device = self.load_device,
 			size           = size,
 		)
 
diff --git a/PixArt/loader.py b/PixArt/loader.py
index 930568a..cedb5fd 100644
--- a/PixArt/loader.py
+++ b/PixArt/loader.py
@@ -115,7 +115,6 @@ def load_pixart(model_path, model_conf=None):
 		model,
 		load_device = load_device,
 		offload_device = offload_device,
-		current_device = "cpu",
 	)
 	return model_patcher
 
diff --git a/PixArt/lora.py b/PixArt/lora.py
index 70fa704..fca5931 100644
--- a/PixArt/lora.py
+++ b/PixArt/lora.py
@@ -78,7 +78,6 @@ def replace_model_patcher(model):
 		size                  = model.size,
 		load_device           = model.load_device,
 		offload_device        = model.offload_device,
-		current_device        = model.current_device,
 		weight_inplace_update = model.weight_inplace_update,
 	)
 	n.patches = {}
diff --git a/T5/loader.py b/T5/loader.py
index f33ef7f..db321ab 100644
--- a/T5/loader.py
+++ b/T5/loader.py
@@ -58,7 +58,6 @@ class EXM_T5v11:
 			self.cond_stage_model,
 			load_device    = self.load_device,
 			offload_device = self.offload_device,
-			current_device = self.load_device,
 			size           = size,
 		)