flax-community
/

t5-vae-wiki

Transformers JAX English transformer_vae vae Inference Endpoints

Model card Files Files and versions Community

Fraser commited on Jul 19, 2021

Commit

b21190b

•

2 Parent(s): 460f960 b38a3ea

merge

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