fukugawa
/

transformer-lm-japanese-0.1b

Text Generation

Inference Endpoints

Model card Files Files and versions Community

fukugawa commited on May 14

Commit

0077f4b

•

1 Parent(s): f7ac145

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -52,10 +52,10 @@ pip install jax==0.4.13
 pip install flax==0.6.11
 pip install sentencepiece==0.1.99
-# For CPU Usage:
 pip install jax[cpu]==0.4.13
-# For GPU Usage:
 pip install --upgrade "jax[cuda12_pip]==0.4.13" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 ```

 pip install flax==0.6.11
 pip install sentencepiece==0.1.99
+# For CPU
 pip install jax[cpu]==0.4.13
+# For GPU
 pip install --upgrade "jax[cuda12_pip]==0.4.13" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 ```