feat: Add VLMConfig to schema for proper config.yaml VLM support

2026-01-07 09:38:45 +08:00
parent bc077475c6
commit f19804facb
3 changed files with 19 additions and 8 deletions
--- a/pixelle_video/config/init.py
+++ b/pixelle_video/config/init.py
@@ -29,7 +29,7 @@ Usage:
    if config_manager.validate():
        print("Config is valid!")
 """
-from .schema import PixelleVideoConfig, LLMConfig, ComfyUIConfig, TTSSubConfig, ImageSubConfig, VideoSubConfig
+from .schema import PixelleVideoConfig, LLMConfig, VLMConfig, ComfyUIConfig, TTSSubConfig, ImageSubConfig, VideoSubConfig
 from .manager import ConfigManager
 from .loader import load_config_dict, save_config_dict

@@ -38,7 +38,8 @@ config_manager = ConfigManager()

 __all__ = [
    "PixelleVideoConfig",
-    "LLMConfig", 
+    "LLMConfig",
+    "VLMConfig",
    "ComfyUIConfig",
    "TTSSubConfig",
    "ImageSubConfig",