Omnigen2 model implementation. (#8669)

2025-06-25 16:35:57 -07:00
parent 7a13f74220
commit ec70ed6aea
13 changed files with 152295 additions and 7 deletions
--- a/comfy/sd1_clip.py
+++ b/comfy/sd1_clip.py
@@ -482,7 +482,8 @@ class SDTokenizer:
            if end_token is not None:
                self.end_token = end_token
            else:
-                self.end_token = empty[0]
+                if has_end_token:
+                    self.end_token = empty[0]

        if pad_token is not None:
            self.pad_token = pad_token