@@ -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
@@ -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
@@ -127,7 +127,6 @@ def __init__(self, no_init=False, device="cpu", dtype=None, model_class="mT5", *
 			self.cond_stage_model,
 			load_device    = self.load_device,
 			offload_device = self.offload_device,
-			current_device = self.load_device,
 			size           = size,
 		)
 
 
@@ -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
 
 
@@ -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 = {}
 
@@ -58,7 +58,6 @@ def __init__(self, textmodel_ver="xxl", embedding_directory=None, textmodel_path
 			self.cond_stage_model,
 			load_device    = self.load_device,
 			offload_device = self.offload_device,
-			current_device = self.load_device,
 			size           = size,
 		)
Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,5 @@ def load_dit(model_path, model_conf):`
`56`	`56`	`model,`
`57`	`57`	`load_device = load_device,`
`58`	`58`	`offload_device = offload_device,`
`59`		`- current_device = "cpu",`
`60`	`59`	`)`
`61`	`60`	`return model_patcher`
Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,5 @@ def load_hydit(model_path, model_conf):`
`76`	`76`	`model,`
`77`	`77`	`load_device = load_device,`
`78`	`78`	`offload_device = offload_device,`
`79`		`- current_device = "cpu",`
`80`	`79`	`)`
`81`	`80`	`return model_patcher`
Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,6 @@ def __init__(self, no_init=False, device="cpu", dtype=None, model_class="mT5", *`
`127`	`127`	`self.cond_stage_model,`
`128`	`128`	`load_device = self.load_device,`
`129`	`129`	`offload_device = self.offload_device,`
`130`		`- current_device = self.load_device,`
`131`	`130`	`size = size,`
`132`	`131`	`)`
`133`	`132`
Original file line number	Diff line number	Diff line change
`@@ -115,7 +115,6 @@ def load_pixart(model_path, model_conf=None):`
`115`	`115`	`model,`
`116`	`116`	`load_device = load_device,`
`117`	`117`	`offload_device = offload_device,`
`118`		`- current_device = "cpu",`
`119`	`118`	`)`
`120`	`119`	`return model_patcher`
`121`	`120`
Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,6 @@ def replace_model_patcher(model):`
`78`	`78`	`size = model.size,`
`79`	`79`	`load_device = model.load_device,`
`80`	`80`	`offload_device = model.offload_device,`
`81`		`- current_device = model.current_device,`
`82`	`81`	`weight_inplace_update = model.weight_inplace_update,`
`83`	`82`	`)`
`84`	`83`	`n.patches = {}`
Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,6 @@ def __init__(self, textmodel_ver="xxl", embedding_directory=None, textmodel_path`
`58`	`58`	`self.cond_stage_model,`
`59`	`59`	`load_device = self.load_device,`
`60`	`60`	`offload_device = self.offload_device,`
`61`		`- current_device = self.load_device,`
`62`	`61`	`size = size,`
`63`	`62`	`)`
`64`	`63`