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,
load_device = load_device,
offload_device = offload_device,
current_device = "cpu",
)
return model_patcher
-1
View File
@@ -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
-1
View File
@@ -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,
)
-1
View File
@@ -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
-1
View File
@@ -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 = {}
-1
View File
@@ -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,
)