ZeroCool94 commited on
Commit
afd479f
1 Parent(s): f8d2192

Uploaded VAE checkpoints for the Muse model.

Browse files

Uploaded VAE checkpoints for the [Muse](https://github.com/lucidrains/muse-maskgit-pytorch) model. For now only the first VAE has been trained as I have not managed to train the base and super res VAE yet, the training script also has issues with the training steps count as it resets after every restart of the training session but for now we are about 100K training steps using the [INE dataset](https://github.com/Sygil-Dev/INE-dataset) and things are starting to show some results on the validation data showing basic shapes and colors on landscape images and even partially showing faces on images containing people.

muse/muse-v0.1-vae.1000.ema.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58d80bd3a33d45c3a6cd3f215299201867228bba83cafd8b8ccc386c521d5521
3
+ size 753966377
muse/muse-v0.1-vae.1000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a40d7502c1f9201d34b193d24a52c0865089fd1165b720013d4f3ce5623702c2
3
+ size 376980822