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

extensions_built_in/diffusion_models/minimax_h3/minimax_h3.py

@@ -1153,6 +1153,9 @@ class MinimaxH3Ref2VAModel(MinimaxH3Model):
         if self.dopsd:
             self.dopsd_self_ref = True
             self.require_pixel_tensor_cache = True
+            self.dopsd_bleed_strength = float(
+                self.model_config.model_kwargs.get("dopsd_bleed_strength", 1.0)
+            )
 
     def _dit_component(self) -> str:
         partition = str(

extensions_built_in/diffusion_models/z_image/z_image.py

@@ -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()