Fraser commited on
Commit
b38a3ea
1 Parent(s): cc547ba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -3
README.md CHANGED
@@ -8,8 +8,6 @@ license: apache-2.0
8
 
9
  A Transformer-VAE made using flax.
10
 
11
- Try the [demo] (TODO)!
12
-
13
  It has been trained to interpolate on sentences form wikipedia.
14
 
15
  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
29
 
30
  tokenizer = AutoTokenizer.from_pretrained("t5-base")
31
 
32
- model = FlaxT5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-python")
33
  ```
34
 
35
  ## Setup
 
8
 
9
  A Transformer-VAE made using flax.
10
 
 
 
11
  It has been trained to interpolate on sentences form wikipedia.
12
 
13
  Done as part of Huggingface community training ([see forum post](https://discuss.huggingface.co/t/train-a-vae-to-interpolate-on-english-sentences/7548)).
 
27
 
28
  tokenizer = AutoTokenizer.from_pretrained("t5-base")
29
 
30
+ model = FlaxT5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-wiki")
31
  ```
32
 
33
  ## Setup