From 2fd51d0d47cf6926537e0175107886c9a7825f6d Mon Sep 17 00:00:00 2001
From: Acly <aclysia@gmail.com>
Date: Sat, 30 May 2026 11:49:19 +0200
Subject: [PATCH] Workaround for import failure in transformers with certain
 pytorch installs #66

* seems to affects pytorch compiled with DISTRIBUTED=0, eg. Windows ROCm
* should probably be fixed in transformers somehow?
---
 __init__.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 08f2e47..cb5c0db 100644
--- a/__init__.py
+++ b/__init__.py
@@ -4,7 +4,7 @@ from . import api as api, nodes, tile, region, nsfw, translation, krita
 
 class ExternalToolingNodes(ComfyExtension):
     async def get_node_list(self) -> list[type[io.ComfyNode]]:
-        return [
+        node_list = [
             nodes.LoadImageCache,
             nodes.SaveImageCache,
             nodes.LoadImageBase64,
@@ -22,7 +22,6 @@ class ExternalToolingNodes(ComfyExtension):
             region.DefineRegion,
             region.ListRegionMasks,
             region.AttentionMask,
-            nsfw.NSFWFilter,
             translation.Translate,
             krita.KritaOutput,
             krita.KritaSendText,
@@ -34,6 +33,17 @@ class ExternalToolingNodes(ComfyExtension):
             krita.KritaStyle,
             krita.KritaStyleAndPrompt,
         ]
+        try:  # see #66
+            import nsfw
+
+            node_list.append(nsfw.NSFWFilter)
+        except (ImportError, ModuleNotFoundError):
+            import traceback
+
+            print("[comfyui-tooling-nodes] WARNING: Could not import all nodes.")
+            traceback.print_exc()
+
+        return node_list
 
 
 async def comfy_entrypoint():