From f6daa7805a83c4ee6b6b971e6f79e0bc9cefa2e3 Mon Sep 17 00:00:00 2001
From: jax <jaxinformationtransformer@gmail.com>
Date: Wed, 2 Apr 2025 17:59:11 +0800
Subject: [PATCH] add hf_token

---
 nodes/comfy_nodes.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/nodes/comfy_nodes.py b/nodes/comfy_nodes.py
index 9696118..f8e9d76 100644
--- a/nodes/comfy_nodes.py
+++ b/nodes/comfy_nodes.py
@@ -12,19 +12,22 @@ from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
 from easycontrol.pipeline import FluxPipeline
 from easycontrol.transformer_flux import FluxTransformer2DModel
 from easycontrol.lora_helper import set_single_lora, set_multi_lora, unset_lora
+from huggingface_hub import login
+
 
 class EasyControlLoadFlux:
     @classmethod
     def INPUT_TYPES(cls):
         return {
-
+            "hf_token": ("STRING", {"default": "", "multiline": True}),
         }
     
     RETURN_TYPES = ("EASYCONTROL_PIPE", "EASYCONTROL_TRANSFORMER")
     FUNCTION = "load_model"
     CATEGORY = "EasyControl"
 
-    def load_model(self):
+    def load_model(self, hf_token):
+        login(token=hf_token)
         base_path = "black-forest-labs/FLUX.1-dev"
         device = "cuda" if torch.cuda.is_available() else "cpu"