1 files changed, 0 insertions, 10 deletions

diff --git a/train_ti.py b/train_ti.py
index da0c03e..7d1ef19 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -698,16 +698,6 @@ def main():
         unet.enable_gradient_checkpointing()
         text_encoder.gradient_checkpointing_enable()
 
-    # convnext = create_model(
-    #     "convnext_tiny",
-    #     pretrained=False,
-    #     num_classes=3,
-    #     drop_path_rate=0.0,
-    # )
-    # convnext.to(accelerator.device, dtype=weight_dtype)
-    # convnext.requires_grad_(False)
-    # convnext.eval()
-
     if len(args.alias_tokens) != 0:
         alias_placeholder_tokens = args.alias_tokens[::2]
         alias_initializer_tokens = args.alias_tokens[1::2]