Update README.md

README.md

@@ -15,7 +15,7 @@ A Deep Learning project that uses Diffusion transformers (DiT) to generate and T
 Please see the GitHub repo for more info: https://github.com/ikergarcia1996/AI-Generated-GTAV
 
 
-### Dit.safetensors
+### dit.safetensors
 - Trained using 4xNvidia A100 80Gb in Bfloat16
 - 64 batch size
 - 1e-4 learning rate with constant scheduler and 5% warmup
@@ -24,8 +24,8 @@ Please see the GitHub repo for more info: https://github.com/ikergarcia1996/AI-G
 - ctx noise increased from 0 to 40 during the first 50% of the training. Set to 40 during the remaining steps.  
 - No Action conditioning
 
-### Dit_actions.safetensors
-- Continual training of Dit.safetensors with action_conditioning for 210,000 steps. 
+### dit_action.safetensors
+- Continue training of `dit.safetensors` with action_conditioning for 210,000 steps. 
 - Trained using 4xNvidia A100 80Gb in Bfloat16
 - 64 batch size
 - 1e-4 learning rate with cosine scheduler to 1e-5 and 5% warmup