fix

README.md

@@ -17,9 +17,9 @@ Builds on T5, using an autoencoder to convert it into an MMD-VAE.
 
 ## How to use from the 🤗/transformers library
 
-Clone the model repo.
+Add model repo as a submodule:
 ```bash
-git clone https://github.com/Fraser-Greenlee/t5-vae-flax.git
+git submodule add https://github.com/Fraser-Greenlee/t5-vae-flax.git t5_vae_flax
 ```
 
 ```python
@@ -28,7 +28,7 @@ from t5_vae_flax.src.t5_vae import FlaxT5VaeForAutoencoding
 
 tokenizer = AutoTokenizer.from_pretrained("t5-base")
 
-model = T5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-python")
+model = FlaxT5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-python")
 ```
 
 ## Setup