From d2b30450e410c8ad047d7c0218b5fe768f71bbd3 Mon Sep 17 00:00:00 2001
From: mayukhdeb <mayukhmainak2000@gmail.com>
Date: Fri, 29 Mar 2024 12:36:06 -0700
Subject: [PATCH] update config args

---
 trainer/config.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/trainer/config.py b/trainer/config.py
index 46b2a14..740fdd3 100644
--- a/trainer/config.py
+++ b/trainer/config.py
@@ -40,13 +40,8 @@ class TrainingConfig(BaseModel, extra = "forbid"):
     debug: bool = False
     hard_pivot: bool = False
     off_ratio_power: float = False
-    num_training_images: int
-    trigger_text: str
-    segmentation_prompt: str
-    training_captions: List[str]
     allow_tf32: bool = True
     mixed_precision: Literal["fp16", "bf16", "fp32"] = "bf16"
-    instance_data_dir: str
     inserting_list_tokens: List[str] = ["<s0>"]
     token_dict: dict = {"TOKEN": "<s0>"}
     device: str = "cuda:0"