Ensure ComfyUI knows about the LLM model path

This commit is contained in:
filtered
2025-02-10 06:24:27 +11:00
parent dffd12506d
commit c5c79d8aa9
+3
View File
@@ -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: