From be12d8d45df033845ca65845ec126f9388c276e0 Mon Sep 17 00:00:00 2001
From: toxicwind <toxicwind@gmail.com>
Date: Sun, 1 Dec 2024 13:24:04 -0700
Subject: [PATCH] Update captures.py

Add support for:
LoraLoaderWithPreviews - https://github.com/X-T-E-R/ComfyUI-EasyCivitai-XTNodes
GGUF - https://github.com/city96/ComfyUI-GGUF/
in capture fields
---
 py/defs/captures.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/py/defs/captures.py b/py/defs/captures.py
index 51eaccb8..d9121baf 100644
--- a/py/defs/captures.py
+++ b/py/defs/captures.py
@@ -93,11 +93,30 @@
         MetaField.LORA_STRENGTH_MODEL: {"field_name": "strength_model"},
         MetaField.LORA_STRENGTH_CLIP: {"value": 0},
     },
+    # LoraLoaderWithPreviews - https://github.com/X-T-E-R/ComfyUI-EasyCivitai-XTNodes
+    "LoraLoaderWithPreviews": {
+        MetaField.LORA_MODEL_NAME: {"field_name": "model_name"},
+        MetaField.LORA_MODEL_HASH: {
+            "field_name": "model_name",
+            "format": calc_lora_hash,
+        },
+        MetaField.LORA_STRENGTH_MODEL: {"field_name": "strength_model"},
+        MetaField.LORA_STRENGTH_CLIP: {"field_name": "strength_clip"},
+    },
     # Flux - https://comfyanonymous.github.io/ComfyUI_examples/flux/
     "UNETLoader": {
         MetaField.MODEL_NAME: {"field_name": "unet_name"},
         MetaField.MODEL_HASH: {"field_name": "unet_name", "format": calc_unet_hash},
     },
+    # GGUF - https://github.com/city96/ComfyUI-GGUF/
+    "UnetLoaderGGUF": {
+        MetaField.MODEL_NAME: {"field_name": "unet_name"},
+        MetaField.MODEL_HASH: {"field_name": "unet_name", "format": calc_unet_hash},
+    },
+    "UnetLoaderGGUFAdvanced": {
+        MetaField.MODEL_NAME: {"field_name": "unet_name"},
+        MetaField.MODEL_HASH: {"field_name": "unet_name", "format": calc_unet_hash},
+    },
     "RandomNoise": {
         MetaField.SEED: {"field_name": "noise_seed"},
     },