[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
|
import torch
|
||||||
from diffusers.image_processor import VaeImageProcessor
|
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 import DiTConfig, VAEConfig
|
||||||
from sglang.multimodal_gen.configs.models.dits.glmimage import GlmImageDitConfig
|
from sglang.multimodal_gen.configs.models.dits.glmimage import GlmImageDitConfig
|
||||||
from sglang.multimodal_gen.configs.models.vaes.glmimage import GlmImageVAEConfig
|
from sglang.multimodal_gen.configs.models.vaes.glmimage import GlmImageVAEConfig
|
||||||
@@ -25,6 +27,10 @@ class GlmImagePipelineConfig(ImagePipelineConfig):
|
|||||||
|
|
||||||
vae_sp: bool = False
|
vae_sp: bool = False
|
||||||
|
|
||||||
|
text_encoder_configs: tuple[EncoderConfig, ...] = field(
|
||||||
|
default_factory=lambda: (T5Config(),)
|
||||||
|
)
|
||||||
|
|
||||||
dit_config: DiTConfig = field(default_factory=GlmImageDitConfig)
|
dit_config: DiTConfig = field(default_factory=GlmImageDitConfig)
|
||||||
# VAE
|
# VAE
|
||||||
vae_config: VAEConfig = field(default_factory=GlmImageVAEConfig)
|
vae_config: VAEConfig = field(default_factory=GlmImageVAEConfig)
|
||||||
|
|||||||
Reference in New Issue
Block a user