From 133d9af042316dddd1d1a530e478b94057f04b02 Mon Sep 17 00:00:00 2001
From: aszc-dev <kontakt@aszczepanski.pl>
Date: Sun, 29 Oct 2023 23:59:54 +0100
Subject: [PATCH] Add sampler to Core ML Suite category

---
 coreml_suite/samplers.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/coreml_suite/samplers.py b/coreml_suite/samplers.py
index 029b6be..84cc1f6 100644
--- a/coreml_suite/samplers.py
+++ b/coreml_suite/samplers.py
@@ -37,6 +37,8 @@ class CoreMLSampler(KSampler):
             "optional": {"latent_image": ("LATENT",)},
         }
 
+    CATEGORY = "CoreML Suite"
+
     def sample(
         self,
         coreml_model,