Stability improvements to model offloading. Added D-OPSD bleed loss as well.

This commit is contained in:
Jaret Burkett
2026-08-29 11:23:47 -06:00
parent 64a20f51a6
commit 683fe8afc0
11 changed files with 390 additions and 45 deletions

View File

@@ -188,6 +188,10 @@ class ZImageModel(BaseModel):
use_comfy_weights=self.model_config.model_kwargs.get(
"use_comfy_weights", True
),
# comfy candidate ranking hint only (aitk_post_load quantizes):
# a matching pre-quantized file loads with no requant work
qtype=self.model_config.qtype if self.model_config.quantize else None,
quantize_on_load=False,
)
flush()