From ac5569f2a04be716aac2d06d9e3d164f19baa090 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 7 Jul 2026 19:46:25 +0800
Subject: [PATCH] Add dataset folder

---
 extra_model_paths.yaml.example | 1 +
 folder_paths.py                | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/extra_model_paths.yaml.example b/extra_model_paths.yaml.example
index 6a31d8a63..755b8d124 100644
--- a/extra_model_paths.yaml.example
+++ b/extra_model_paths.yaml.example
@@ -29,6 +29,7 @@
 #     upscale_models: models/upscale_models/
 #     latent_upscale_models: models/latent_upscale_models/
 #     custom_nodes: custom_nodes/
+#     datasets: datasets/
 #     hypernetworks: models/hypernetworks/
 #     photomaker: models/photomaker/
 #     classifiers: models/classifiers/
diff --git a/folder_paths.py b/folder_paths.py
index ee048b0f2..c91065ccc 100644
--- a/folder_paths.py
+++ b/folder_paths.py
@@ -40,6 +40,8 @@ folder_names_and_paths["latent_upscale_models"] = ([os.path.join(models_dir, "la
 
 folder_names_and_paths["custom_nodes"] = ([os.path.join(base_path, "custom_nodes")], set())
 
+folder_names_and_paths["datasets"] = ([os.path.join(base_path, "datasets")], set())
+
 folder_names_and_paths["hypernetworks"] = ([os.path.join(models_dir, "hypernetworks")], supported_pt_extensions)
 
 folder_names_and_paths["photomaker"] = ([os.path.join(models_dir, "photomaker")], supported_pt_extensions)