@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -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
|
||||||
|
|
||||||
|
|||||||
@@ -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 = {}
|
||||||
|
|||||||
@@ -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,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user