wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-24 09:22:32 +08:00
Code Issues Actions 24 Packages Projects Releases Wiki Activity

Move cogvideo text encoder into a dedicated module.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-04-14 14:58:37 +02:00
parent 40aa594d29
commit 191cda4dc4
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/supported_models.py
View File

@ -27,6 +27,7 @@ import comfy.text_encoders.anima
import comfy.text_encoders.ace15 import comfy.text_encoders.ace15
import comfy.text_encoders.longcat_image import comfy.text_encoders.longcat_image
import comfy.text_encoders.ernie import comfy.text_encoders.ernie
import comfy.text_encoders.cogvideo
from . import supported_models_base from . import supported_models_base
from . import latent_formats from . import latent_formats
@ -1862,11 +1863,7 @@ class CogVideoX_T2V(supported_models_base.BASE):
return out return out
def clip_target(self, state_dict={}): def clip_target(self, state_dict={}):
class CogVideoXT5Tokenizer(comfy.text_encoders.sd3_clip.T5XXLTokenizer): return supported_models_base.ClipTarget(comfy.text_encoders.cogvideo.CogVideoXT5Tokenizer, comfy.text_encoders.sd3_clip.T5XXLModel)
def __init__(self, embedding_directory=None, tokenizer_data={}):
super().__init__(embedding_directory=embedding_directory, tokenizer_data=tokenizer_data, min_length=226)
return supported_models_base.ClipTarget(CogVideoXT5Tokenizer, comfy.text_encoders.sd3_clip.T5XXLModel)
class CogVideoX_I2V(CogVideoX_T2V): class CogVideoX_I2V(CogVideoX_T2V):
unet_config = { unet_config = {

6
comfy/text_encoders/cogvideo.py Normal file
View File

@ -0,0 +1,6 @@
import comfy.text_encoders.sd3_clip
class CogVideoXT5Tokenizer(comfy.text_encoders.sd3_clip.T5XXLTokenizer):
def __init__(self, embedding_directory=None, tokenizer_data={}):
super().__init__(embedding_directory=embedding_directory, tokenizer_data=tokenizer_data, min_length=226)