[Bug fix] use default_factory to provide default value in dataclasses…

… in energon MultiModalSampleConfig (NVIDIA#11041) Signed-off-by: Guyue Huang <[email protected]> Co-authored-by: Guyue Huang <[email protected]>
paarthneekhara · Jan 18, 2025 · dab3116 · dab3116
1 parent a89e0e8
commit dab3116
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/multimodal/data/energon/config.py b/nemo/collections/multimodal/data/energon/config.py
@@ -64,7 +64,7 @@ class LLaVATemplateConfig(BaseConversationTemplateConfig):
 
 @dataclass
 class MultiModalSampleConfig:
-    image_token: ImageToken = ImageToken()
+    image_token: ImageToken = field(default_factory=ImageToken)
     ignore_place_holder: int = -100
-    conversation_template_config: LLaVATemplateConfig = LLaVATemplateConfig()
+    conversation_template_config: LLaVATemplateConfig = field(default_factory=LLaVATemplateConfig)
     image_following_text: bool = True