PixArt: Update import for SD3 clip

#76
2024-07-17 00:22:24 +02:00
parent 8cff976421
commit 1db046de78
1 changed files with 5 additions and 1 deletions
@@ -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()