it's Friday night, let cross finger

huggingface · Dec 13, 2024 · 2c47915 · 2c47915
1 parent 9c8aff7
commit 2c47915
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/kosmos2_5/configuration_kosmos2_5.py b/src/transformers/models/kosmos2_5/configuration_kosmos2_5.py
@@ -268,6 +268,7 @@ class Kosmos2_5Config(PretrainedConfig):
 
     model_type = "kosmos-2.5"
     is_composition = True
+    sub_configs = {"text_config": Kosmos2_5TextConfig, "vision_config": Kosmos2_5VisionConfig}
 
     def __init__(
         self,