readme

lucidrains · Jun 18, 2024 · 495d3d4 · 495d3d4
1 parent 479b8f8
commit 495d3d4
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -4,11 +4,17 @@
 
 Implementation of TiTok, proposed by Bytedance in <a href="https://yucornetto.github.io/projects/titok.html">An Image is Worth 32 Tokens for Reconstruction and Generation</a>
 
+## Install
+
+```bash
+$ pip install titok-pytorch
+```
+
 ## Usage
 
 ```python
 import torch
-from titok_pytorch.titok import TiTokTokenizer
+from titok_pytorch import TiTokTokenizer
 
 images = torch.randn(2, 3, 256, 256)