Remove unused arg

#80
This commit is contained in:
City
2024-08-07 13:42:31 +02:00
parent 7e96bf676c
commit 193c2fd9d3
6 changed files with 0 additions and 6 deletions
-1
View File
@@ -56,6 +56,5 @@ def load_dit(model_path, model_conf):
model, model,
load_device = load_device, load_device = load_device,
offload_device = offload_device, offload_device = offload_device,
current_device = "cpu",
) )
return model_patcher return model_patcher
-1
View File
@@ -76,6 +76,5 @@ def load_hydit(model_path, model_conf):
model, model,
load_device = load_device, load_device = load_device,
offload_device = offload_device, offload_device = offload_device,
current_device = "cpu",
) )
return model_patcher return model_patcher
-1
View File
@@ -127,7 +127,6 @@ class EXM_HyDiT_Tenc_Temp:
self.cond_stage_model, self.cond_stage_model,
load_device = self.load_device, load_device = self.load_device,
offload_device = self.offload_device, offload_device = self.offload_device,
current_device = self.load_device,
size = size, size = size,
) )
-1
View File
@@ -115,7 +115,6 @@ def load_pixart(model_path, model_conf=None):
model, model,
load_device = load_device, load_device = load_device,
offload_device = offload_device, offload_device = offload_device,
current_device = "cpu",
) )
return model_patcher return model_patcher
-1
View File
@@ -78,7 +78,6 @@ def replace_model_patcher(model):
size = model.size, size = model.size,
load_device = model.load_device, load_device = model.load_device,
offload_device = model.offload_device, offload_device = model.offload_device,
current_device = model.current_device,
weight_inplace_update = model.weight_inplace_update, weight_inplace_update = model.weight_inplace_update,
) )
n.patches = {} n.patches = {}
-1
View File
@@ -58,7 +58,6 @@ class EXM_T5v11:
self.cond_stage_model, self.cond_stage_model,
load_device = self.load_device, load_device = self.load_device,
offload_device = self.offload_device, offload_device = self.offload_device,
current_device = self.load_device,
size = size, size = size,
) )