HunYuanDiT Fix text encoder device select

Fixes #47 and #50
2024-05-25 16:44:58 +02:00
parent 1953a3bc99
commit e5c15ba479
3 changed files with 17 additions and 11 deletions
@@ -73,5 +73,6 @@ def load_hydit(model_path, model_conf):
 		load_device = load_device,
 		offload_device = offload_device,
 		current_device = "cpu",
+		size = 6 * (1024**3),
 	)
 	return model_patcher