flax-community
/

t5-vae-wiki

transformer_vae

Model card Files Files and versions

Fraser commited on Jul 19, 2021

Commit

b38a3ea

·

1 Parent(s): cc547ba

Update README.md

Files changed (1) hide show

README.md +1 -3

README.md CHANGED Viewed

@@ -8,8 +8,6 @@ license: apache-2.0
 A Transformer-VAE made using flax.
-Try the [demo] (TODO)!
 It has been trained to interpolate on sentences form wikipedia.
 Done as part of Huggingface community training ([see forum post](https://discuss.huggingface.co/t/train-a-vae-to-interpolate-on-english-sentences/7548)).
@@ -29,7 +27,7 @@ from t5_vae_flax.src.t5_vae import FlaxT5VaeForAutoencoding
 tokenizer = AutoTokenizer.from_pretrained("t5-base")
-model = FlaxT5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-python")
 ```
 ## Setup

 A Transformer-VAE made using flax.
 It has been trained to interpolate on sentences form wikipedia.
 Done as part of Huggingface community training ([see forum post](https://discuss.huggingface.co/t/train-a-vae-to-interpolate-on-english-sentences/7548)).
 tokenizer = AutoTokenizer.from_pretrained("t5-base")
+model = FlaxT5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-wiki")
 ```
 ## Setup