diff --git a/README.md b/README.md
index 12dacdd..8aaa8c8 100755
--- a/README.md
+++ b/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
 
diff --git a/requirements.txt b/requirements.txt
index 2845ff5..ee01c4f 100644
--- a/requirements.txt
+++ b/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
diff --git a/train_configs/training_args_face_sdxl.json b/train_configs/training_args_face_sdxl.json
index 3662f3d..a469f91 100644
--- a/train_configs/training_args_face_sdxl.json
+++ b/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,