Update README.md

README.md

@@ -47,7 +47,7 @@ This model is still in its infancy and it's meant to be constantly updated and t
     - [vae.sygil_muse_v0.7.pt](https://huggingface.co/Sygil/Sygil-Muse/blob/main/vae.sygil_muse_v0.7.pt): Trained from scratch for 440K steps with **dim: 128**, **vq_codebook_dim 4096** and **vq_codebook_size: 16384**.
     - [maskgit.sygil_muse_v0.7.pt](https://huggingface.co/Sygil/Sygil-Muse/blob/main/maskgit.sygil_muse_v0.7.pt): Maskgit trained from the **vae.sygil_muse_v0.7.pt** VAE for 108K steps.
   - #### Beta:
-    - [vae.832000.pt](https://huggingface.co/Sygil/Sygil-Muse/blob/main/vae.832000.pt): Trained from scratch for 832K steps with **dim: 32**,**vq_codebook_dim: 8192** and **vq_codebook_size: 8192**.
+    - [vae.1076000.pt](https://huggingface.co/Sygil/Sygil-Muse/blob/main/vae.1076000.pt): Trained from scratch for 1.07M steps with **dim: 32**,**vq_codebook_dim: 8192** and **vq_codebook_size: 8192**.
   
   Note: Checkpoints under the Beta section are updated daily or at least 3-4 times a week. While the beta checkpoints can be used as they are, only the latest version is kept on the repo and the older checkpoints are removed when a new one
   is uploaded to keep the repo clean.
@@ -76,7 +76,8 @@ The model was trained on the following dataset:
 - **heads:** 8
 - **depth:** 4
 - **Random Crop:** True
-- **Total Training Steps:** 832,000
+- **Total MaskGit Training Steps:** 203,000
+- **Total VAE Training Steps:** 1,076,000
 
   Note: On Muse we can change the image_size or resolution at any time without having to train the model from scratch again, this allows us to first train the model at low resolution using the same `dim` and `vq_codebook_size` to train faster and then we can increase the `image_size` and use a higher resolution once the model has trained enough.