Update README.md

README.md

@@ -10,6 +10,9 @@ This network uses the optimal transport flow matching objective outlined [Flow M
 xATGLU Layers are used instead of linears for entry into the transformer MLP layer [Expanded Gating Ranges
 Improve Activation Functions](https://arxiv.org/pdf/2405.20768)
 
+`python train.py` will train a new image network on the provided dataset. A pretrained model is provided here, step_1799.safetensors. To test, use:
+`python test_sample.py step_1799.safetensors`
+
 ![samples](./1.png)
 ![samples](./2.png)
 ![samples](./3.png)