From c5c79d8aa92cfb0c415b748d63a8a7d6c1f178c4 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Mon, 10 Feb 2025 06:24:27 +1100
Subject: [PATCH] Ensure ComfyUI knows about the LLM model path

---
 nodes.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nodes.py b/nodes.py
index cac283f..3ee5308 100644
--- a/nodes.py
+++ b/nodes.py
@@ -36,6 +36,9 @@ script_directory = os.path.dirname(os.path.abspath(__file__))
 model_directory = os.path.join(folder_paths.models_dir, "LLM")
 os.makedirs(model_directory, exist_ok=True)
 
+# Ensure ComfyUI knows about the LLM model path
+folder_paths.add_model_folder_path("LLM", model_directory)
+
 from transformers import AutoModelForCausalLM, AutoProcessor, set_seed
 
 class DownloadAndLoadFlorence2Model: