From febf73f04d71aede8040328c7bc340180b5985c9 Mon Sep 17 00:00:00 2001
From: Raykosan <71210053+Raykosan@users.noreply.github.com>
Date: Sun, 7 Jun 2026 17:05:04 +0300
Subject: [PATCH] Modification of the RS Models Loader node: pixel_spaces has
 been added to the VAE list

---
 Rayko_Models_Loader.py | 18 +++++++++++++++---
 __init__.py            |  2 +-
 package.json           |  2 +-
 pyproject.toml         |  2 +-
 4 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/Rayko_Models_Loader.py b/Rayko_Models_Loader.py
index e1e3dfa..cc866c3 100644
--- a/Rayko_Models_Loader.py
+++ b/Rayko_Models_Loader.py
@@ -15,6 +15,7 @@
 
 import folder_paths
 import comfy.sd
+import torch
 import json
 import os
 from nodes import UNETLoader, CLIPLoader, VAELoader, LoraLoader, DualCLIPLoader
@@ -27,6 +28,9 @@ class RaykoModelsLoader:
         unet_files = folder_paths.get_filename_list("unet")
         clip_files = folder_paths.get_filename_list("clip")
         vae_files = folder_paths.get_filename_list("vae")
+        
+        # Добавляем "pixel_space" в список VAE (как в нативной ноде ComfyUI)
+        vae_files_with_pixel = vae_files + ["pixel_space"]
 
         weight_dtype_opts = ["default", "fp8_e4m3fn", "fp8_e4m3fn_fast", "fp8_e5m2"]
         
@@ -44,7 +48,7 @@ class RaykoModelsLoader:
             "clip_name2": (clip_files, {"tooltip": "Second CLIP model (for dual-clip models like Flux, SD3)"}),
             "clip_type": (clip_type_opts, {"default": "stable_diffusion"}),
             "clip_device": (device_opts, {"default": "default"}),
-            "vae_name": (vae_files, {"tooltip": "model VAE"}),
+            "vae_name": (vae_files_with_pixel, {"tooltip": "model VAE (includes pixel_space for direct pixel manipulation)"}),
             "lora_data": ("STRING", {"default": "[]", "multiline": False, "forceInput": False}),
         }
         return {"required": required}
@@ -99,8 +103,16 @@ class RaykoModelsLoader:
             )[0]
         print(f"[Rayko] CLIP loaded successfully")
 
-        vae = VAELoader().load_vae(vae_name=vae_name)[0]
-        print(f"[Rayko] VAE: {vae_name}")
+        # VAE loading logic with pixel_space support
+        if vae_name == "pixel_space":
+            print(f"[Rayko] Using pixel_space VAE (fake VAE for direct pixel manipulation)")
+            sd = {}
+            sd["pixel_space_vae"] = torch.tensor(1.0)
+            vae = comfy.sd.VAE(sd=sd, metadata=None)
+            vae.throw_exception_if_invalid()
+        else:
+            vae = VAELoader().load_vae(vae_name=vae_name)[0]
+            print(f"[Rayko] VAE: {vae_name}")
 
         try:
             loras = json.loads(lora_data) if lora_data else []
diff --git a/__init__.py b/__init__.py
index 0a7cef0..cdc1bfa 100644
--- a/__init__.py
+++ b/__init__.py
@@ -29,7 +29,7 @@ IGNORED_MODULES: Set[str] = {'__init__.py'}
 REQUIRED_MODULES: Set[str] = set()
 
 PACKAGE_DISPLAY_NAME = "ComfyUI_RaykoStudio"
-__version__ = "0.21.1"
+__version__ = "0.21.2"
 
 def load_modules():
     print(f"🦊 {PACKAGE_DISPLAY_NAME}: Version: {__version__}")
diff --git a/package.json b/package.json
index cd8422c..4760ad6 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "ComfyUI_RaykoStudio",
-  "version": "0.21.1",
+  "version": "0.21.2",
   "description": "Set of custom nodes for ComfyUI providing additional image processing capabilities",
   "author": "Raykosan",
   "license": "Apache License 2.0",
diff --git a/pyproject.toml b/pyproject.toml
index 9f6add0..c4f7bb2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "ComfyUI_RaykoStudio"
 description = "Set of custom nodes for ComfyUI providing additional image processing capabilities"
-version = "0.21.1"
+version = "0.21.2"
 license = { file = "LICENSE" }
 dependencies = ["torch>=1.7.0", "numpy>=1.19.0", "Pillow>=8.0.0", "freetype-py>=2.5.1"]