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