[update] add requirements.txt

FoundationVision · Dec 24, 2024 · a03f376 · a03f376
1 parent fac801d
commit a03f376
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -35,7 +35,7 @@ We also provide [interactive_infer.ipynb](tools/interactive_infer.ipynb) for you
   - [x] Web Demo 
   - [x] Inference Code
   - [x] Infinity-2B Checkpoints
-  - [x] VAE Checkpoints
+  - [x] Visual Tokenizer Checkpoints
 
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,21 @@
+easydict
+typed-argument-parser
+seaborn
+kornia
+gputil
+colorama
+omegaconf
+pandas
+timm==0.9.6
+decord
+transformers
+torch==2.5.1
+pytz
+pandas
+wandb
+colorama
+imageio
+einops
+openai
+httpx==0.20.0
+opencv-python