push changes

README.md

@@ -44,8 +44,9 @@ sudo curl -o /usr/local/bin/cog -L "https://github.com/replicate/cog/releases/la
 sudo chmod +x /usr/local/bin/cog
 ```
 
-2. Build the image with `sudo cog build`
-3. Run a training run with `sudo sh cog_test_train.sh`
+2. Build the image with `cog build`
+3. Run a training run with `sh cog_test_train.sh`
+4. You can also go into the container with `cog run /bin/bash`
 
 ## Automatic Checkpoint Evaluation
 

requirements.txt

@@ -12,7 +12,7 @@ invisible-watermark==0.2.0
 pandas==2.2.1
 numpy==1.26.4
 opencv-python==4.10.0.84
-mediapipe== 0.10.14
+mediapipe==0.10.14
 openai==1.35.13
 python-dotenv==1.0.1
 prodigyopt==1.0

train_configs/training_args_face_sdxl.json

@@ -13,7 +13,7 @@
     "ti_lr": 0.001,
     "ti_weight_decay": 0.0005,
 
-    "remove_ti_token_from_prompts": false,
+    "disable_ti": false,
     "text_encoder_lora_optimizer": null,
     "text_encoder_lora_lr": 1.0e-4,
     "text_encoder_lora_weight_decay": 1e-5,