Update trainer.py

trainer.py

@@ -1,10 +1,14 @@
 # =============================================================================
 # Cell 1: Install dependencies
 # =============================================================================
-# !pip install -q torch torchvision safetensors transformers datasets
-# !pip install -q git+https://github.com/AbstractEyes/lattice_vocabulary.git
-# !cd /content && git clone https://github.com/AbstractPhil/sd15-trainer-geo.git
-# !cd /content/sd15-trainer-geo && pip install -e .
+"""
+try:
+  !pip uninstall -qy sd15-flow-trainer[dev]
+except:
+  pass
+
+!pip install "sd15-flow-trainer[dev] @ git+https://github.com/AbstractEyes/sd15-flow-trainer.git" -q
+"""
 
 # =============================================================================
 # Cell 2: Pre-encode 50k object-relations dataset