[Diffusion] [BUG] Fix missing initialization of GLM-Image text encoder config (#18704)
Co-authored-by: Hao Jin <Hao Jin>
This commit is contained in:
@@ -3,6 +3,8 @@ from dataclasses import dataclass, field
|
||||
import torch
|
||||
from diffusers.image_processor import VaeImageProcessor
|
||||
|
||||
from python.sglang.multimodal_gen.configs.models.encoders.base import EncoderConfig
|
||||
from python.sglang.multimodal_gen.configs.models.encoders.t5 import T5Config
|
||||
from sglang.multimodal_gen.configs.models import DiTConfig, VAEConfig
|
||||
from sglang.multimodal_gen.configs.models.dits.glmimage import GlmImageDitConfig
|
||||
from sglang.multimodal_gen.configs.models.vaes.glmimage import GlmImageVAEConfig
|
||||
@@ -25,6 +27,10 @@ class GlmImagePipelineConfig(ImagePipelineConfig):
|
||||
|
||||
vae_sp: bool = False
|
||||
|
||||
text_encoder_configs: tuple[EncoderConfig, ...] = field(
|
||||
default_factory=lambda: (T5Config(),)
|
||||
)
|
||||
|
||||
dit_config: DiTConfig = field(default_factory=GlmImageDitConfig)
|
||||
# VAE
|
||||
vae_config: VAEConfig = field(default_factory=GlmImageVAEConfig)
|
||||
|
||||
Reference in New Issue
Block a user