From b2593317bfedd7ad4f5bd2df680e1258053518d6 Mon Sep 17 00:00:00 2001
From: Mihail Karaev <karaevmihail@gmail.com>
Date: Mon, 12 Jan 2026 12:17:25 +0000
Subject: [PATCH] Change dit format to bf16

---
 comfy/supported_models.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/comfy/supported_models.py b/comfy/supported_models.py
index ad3c3b40b..1f928bb49 100644
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -1535,7 +1535,7 @@ class Kandinsky5Image(Kandinsky5):
         hunyuan_detect = comfy.text_encoders.hunyuan_video.llama_detect(state_dict, "{}qwen25_7b.transformer.".format(pref))
         return supported_models_base.ClipTarget(comfy.text_encoders.kandinsky5.Kandinsky5TokenizerImage, comfy.text_encoders.kandinsky5.te(**hunyuan_detect))
 
-class Kandinsky5ImageToImage(supported_models_base.BASE):
+class Kandinsky5ImageToImage(Kandinsky5):
     unet_config = {
         "image_model": "kandinsky5",
         "model_dim": 2560,
@@ -1545,7 +1545,6 @@ class Kandinsky5ImageToImage(supported_models_base.BASE):
     sampling_settings = {
         "shift": 3.0,
     }
-
     latent_format = latent_formats.Flux
     memory_usage_factor = 1.25 #TODO