Spaces:
Runtime error
Runtime error
Nef Caballero
commited on
Commit
·
be70f8e
1
Parent(s):
b061f17
fix attempt for HG error 2
Browse files
app.py
CHANGED
|
@@ -100,7 +100,6 @@ def add_extra_model_paths() -> None:
|
|
| 100 |
add_comfyui_directory_to_sys_path()
|
| 101 |
add_extra_model_paths()
|
| 102 |
|
| 103 |
-
|
| 104 |
def import_custom_nodes() -> None:
|
| 105 |
"""Find all custom nodes in the custom_nodes folder and add those node objects to NODE_CLASS_MAPPINGS
|
| 106 |
This function sets up a new asyncio event loop, initializes the PromptServer,
|
|
@@ -122,18 +121,20 @@ def import_custom_nodes() -> None:
|
|
| 122 |
# Initializing custom nodes
|
| 123 |
init_extra_nodes()
|
| 124 |
|
|
|
|
|
|
|
| 125 |
|
|
|
|
| 126 |
from nodes import NODE_CLASS_MAPPINGS
|
| 127 |
|
| 128 |
-
|
| 129 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 130 |
|
| 131 |
-
#To be added to `model_loaders` as it loads a model
|
| 132 |
-
dualcliploader_357 = dualcliploader.load_clip(
|
| 133 |
-
clip_name1="t5/t5xxl_fp16.safetensors",
|
| 134 |
-
clip_name2="clip_l.safetensors",
|
| 135 |
-
type="flux",
|
| 136 |
-
)
|
| 137 |
cr_clip_input_switch = NODE_CLASS_MAPPINGS["CR Clip Input Switch"]()
|
| 138 |
cliptextencode = NODE_CLASS_MAPPINGS["CLIPTextEncode"]()
|
| 139 |
loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
|
|
@@ -168,7 +169,7 @@ downloadandloaddepthanythingv2model_437 = (
|
|
| 168 |
instructpixtopixconditioning = NODE_CLASS_MAPPINGS[
|
| 169 |
"InstructPixToPixConditioning"
|
| 170 |
]()
|
| 171 |
-
text_multiline_454 =
|
| 172 |
clipvisionloader = NODE_CLASS_MAPPINGS["CLIPVisionLoader"]()
|
| 173 |
|
| 174 |
#To be added to `model_loaders` as it loads a model
|
|
@@ -197,7 +198,11 @@ saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
|
|
| 197 |
imagecrop = NODE_CLASS_MAPPINGS["ImageCrop+"]()
|
| 198 |
|
| 199 |
#Add all the models that load a safetensors file
|
| 200 |
-
model_loaders = [
|
|
|
|
|
|
|
|
|
|
|
|
|
| 201 |
|
| 202 |
# Check which models are valid and how to best load them
|
| 203 |
valid_models = [
|
|
@@ -211,7 +216,6 @@ model_management.load_models_gpu(valid_models)
|
|
| 211 |
|
| 212 |
@spaces.GPU(duration=60)
|
| 213 |
def generate_image(prompt, structure_image, style_image, depth_strength, style_strength):
|
| 214 |
-
import_custom_nodes()
|
| 215 |
with torch.inference_mode():
|
| 216 |
|
| 217 |
intconstant_83 = intconstant.get_value(value=1024)
|
|
@@ -220,8 +224,16 @@ def generate_image(prompt, structure_image, style_image, depth_strength, style_s
|
|
| 220 |
|
| 221 |
cr_clip_input_switch_319 = cr_clip_input_switch.switch(
|
| 222 |
Input=1,
|
| 223 |
-
clip1=get_value_at_index(
|
| 224 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 225 |
)
|
| 226 |
|
| 227 |
cliptextencode_174 = cliptextencode.encode(
|
|
|
|
| 100 |
add_comfyui_directory_to_sys_path()
|
| 101 |
add_extra_model_paths()
|
| 102 |
|
|
|
|
| 103 |
def import_custom_nodes() -> None:
|
| 104 |
"""Find all custom nodes in the custom_nodes folder and add those node objects to NODE_CLASS_MAPPINGS
|
| 105 |
This function sets up a new asyncio event loop, initializes the PromptServer,
|
|
|
|
| 121 |
# Initializing custom nodes
|
| 122 |
init_extra_nodes()
|
| 123 |
|
| 124 |
+
# Initialize nodes before using them
|
| 125 |
+
import_custom_nodes()
|
| 126 |
|
| 127 |
+
# Now import and use NODE_CLASS_MAPPINGS
|
| 128 |
from nodes import NODE_CLASS_MAPPINGS
|
| 129 |
|
| 130 |
+
try:
|
| 131 |
+
intconstant = NODE_CLASS_MAPPINGS["INTConstant"]()
|
| 132 |
+
dualcliploader = NODE_CLASS_MAPPINGS["DualCLIPLoader"]()
|
| 133 |
+
except KeyError as e:
|
| 134 |
+
print(f"Error: Could not find node {e} in NODE_CLASS_MAPPINGS")
|
| 135 |
+
print("Available nodes:", list(NODE_CLASS_MAPPINGS.keys()))
|
| 136 |
+
raise
|
| 137 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 138 |
cr_clip_input_switch = NODE_CLASS_MAPPINGS["CR Clip Input Switch"]()
|
| 139 |
cliptextencode = NODE_CLASS_MAPPINGS["CLIPTextEncode"]()
|
| 140 |
loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
|
|
|
|
| 169 |
instructpixtopixconditioning = NODE_CLASS_MAPPINGS[
|
| 170 |
"InstructPixToPixConditioning"
|
| 171 |
]()
|
| 172 |
+
text_multiline_454 = NODE_CLASS_MAPPINGS["Text Multiline"].text_multiline(text="FLUX_Redux")
|
| 173 |
clipvisionloader = NODE_CLASS_MAPPINGS["CLIPVisionLoader"]()
|
| 174 |
|
| 175 |
#To be added to `model_loaders` as it loads a model
|
|
|
|
| 198 |
imagecrop = NODE_CLASS_MAPPINGS["ImageCrop+"]()
|
| 199 |
|
| 200 |
#Add all the models that load a safetensors file
|
| 201 |
+
model_loaders = [dualcliploader.load_clip(
|
| 202 |
+
clip_name1="t5/t5xxl_fp16.safetensors",
|
| 203 |
+
clip_name2="clip_l.safetensors",
|
| 204 |
+
type="flux",
|
| 205 |
+
), vaeloader_359, unetloader_358, clipvisionloader_438, stylemodelloader_441, downloadandloaddepthanythingv2model_437]
|
| 206 |
|
| 207 |
# Check which models are valid and how to best load them
|
| 208 |
valid_models = [
|
|
|
|
| 216 |
|
| 217 |
@spaces.GPU(duration=60)
|
| 218 |
def generate_image(prompt, structure_image, style_image, depth_strength, style_strength):
|
|
|
|
| 219 |
with torch.inference_mode():
|
| 220 |
|
| 221 |
intconstant_83 = intconstant.get_value(value=1024)
|
|
|
|
| 224 |
|
| 225 |
cr_clip_input_switch_319 = cr_clip_input_switch.switch(
|
| 226 |
Input=1,
|
| 227 |
+
clip1=get_value_at_index(dualcliploader.load_clip(
|
| 228 |
+
clip_name1="t5/t5xxl_fp16.safetensors",
|
| 229 |
+
clip_name2="clip_l.safetensors",
|
| 230 |
+
type="flux",
|
| 231 |
+
), 0),
|
| 232 |
+
clip2=get_value_at_index(dualcliploader.load_clip(
|
| 233 |
+
clip_name1="t5/t5xxl_fp16.safetensors",
|
| 234 |
+
clip_name2="clip_l.safetensors",
|
| 235 |
+
type="flux",
|
| 236 |
+
), 0),
|
| 237 |
)
|
| 238 |
|
| 239 |
cliptextencode_174 = cliptextencode.encode(
|