From 1db046de78bd543c895253f71f41c7210a8dc58e Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Wed, 17 Jul 2024 00:22:24 +0200
Subject: [PATCH] PixArt: Update import for SD3 clip

#76
---
 PixArt/nodes.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/PixArt/nodes.py b/PixArt/nodes.py
index ba4d0ee..414e117 100644
--- a/PixArt/nodes.py
+++ b/PixArt/nodes.py
@@ -228,7 +228,11 @@ class PixArtT5FromSD3CLIP:
 	TITLE = "PixArt T5 from SD3 CLIP"
 
 	def split(self, sd3_clip, padding):
-		from comfy.sd3_clip import SD3Tokenizer, SD3ClipModel
+		try:
+			from comfy.text_encoders.sd3_clip import SD3Tokenizer, SD3ClipModel
+		except ImportError:
+			# fallback for older ComfyUI versions
+			from comfy.sd3_clip import SD3Tokenizer, SD3ClipModel
 		import copy
 	
 		clip = sd3_clip.clone()