Update README.md

Support for schnell-compat / svdquant

.github/workflows/publish_action.yml

@@ -0,0 +1,24 @@
+name: Publish to Comfy registry
+on:
+  workflow_dispatch:
+  push:
+    branches:
+      - main
+    paths:
+      - "pyproject.toml"
+
+permissions:
+  issues: write
+
+jobs:
+  publish-node:
+    name: Publish Custom Node to registry
+    runs-on: ubuntu-latest
+    if: ${{ github.repository_owner == 'lodestone-rock' }}
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+      - name: Publish Custom Node
+        uses: Comfy-Org/publish-node-action@v1
+        with:
+          personal_access_token: ${{ secrets.POTATO }} ## Add your own personal access token to your Github Repository secrets and reference it 

README.md

@@ -1,5 +1,6 @@
-# Chroma Manual [WIP]
+# DEPRECATED! Chroma Manual [WIP]
 Chroma is a 8.9B parameter model based on FLUX.1-schnell.
+use comfyui default workflow here instead [https://comfyanonymous.github.io/ComfyUI_examples/chroma/](https://comfyanonymous.github.io/ComfyUI_examples/chroma/)
 
 ## Requirements
 

flux_mod/loader.py

@@ -22,7 +22,10 @@ def ensure_gguf():
     if gguf:
         return
     import sys
-    gguf = sys.modules.get("ComfyUI-GGUF")
+    gguf = next(
+        (mod for path, mod in sys.modules.items() if path.endswith("ComfyUI-GGUF")),
+        None
+    )
     if gguf is None:
         raise RuntimeError("Could not find ComfyUI-GGUF node: GGUF support requires ComfyUI-GGUF")
 

flux_mod/nodes.py

@@ -124,6 +124,8 @@ class ChromaPromptTruncation:
     def append(self, conditioning):
         pruning_idx = conditioning[0][1]["attention_mask"].sum() + 1
         conditioning[0][0] = conditioning[0][0][:, :pruning_idx]
+        del conditioning[0][1]["attention_mask"]
+        conditioning[0][1]["pooled_output"] = torch.zeros(1, 768, dtype=torch.float32)
         c = node_helpers.conditioning_set_values(conditioning, {"guidance": 0})
         return (c,)
 

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
-name = "comfyui_fluxmod"
-description = ""
-version = "1.0.0"
+name = "ComfyUI_FluxMod"
+description = "Nodes for use of Chroma model and other prototype models"
+version = "1.0.1"
 license = {file = "LICENSE"}
 
 [project.urls]
@@ -9,6 +9,6 @@ Repository = "https://github.com/lodestone-rock/ComfyUI_FluxMod"
 #  Used by Comfy Registry https://comfyregistry.org
 
 [tool.comfy]
-PublisherId = "horizon"
+PublisherId = "lodestone-rock"
 DisplayName = "ComfyUI_FluxMod"
 Icon = ""