Added this not that guidance. Added ability to replace prompts.

toolkit/config_modules.py

@@ -496,6 +496,8 @@ class DatasetConfig:
         self.clip_image_path: str = kwargs.get('clip_image_path', None)  # depth maps, etc
         self.clip_image_augmentations: List[dict] = kwargs.get('clip_image_augmentations', None)
         self.clip_image_shuffle_augmentations: bool = kwargs.get('clip_image_shuffle_augmentations', False)
+        self.replacements: List[str] = kwargs.get('replacements', [])
+        self.loss_multiplier: float = kwargs.get('loss_multiplier', 1.0)
 
 
 def preprocess_dataset_raw_config(raw_config: List[dict]) -> List[dict]: