Add reqs and preload

README.md

@@ -6,4 +6,9 @@ colorFrom: red
 colorTo: red
 pinned: true
 short_description: Generate images with OpenTVAR
----
+preload_from_hub:
+  - michellemoorre/vae-test
+  - michellemoorre/vae-test
+  - openai/clip-vit-large-patch14
+  - laion/CLIP-ViT-bigG-14-laion2B-39B-b160k
+---

requirements.txt

@@ -0,0 +1,24 @@
+apex==0.9.10dev
+decord==0.6.0
+einops==0.8.0
+flash_attn==2.6.3
+huggingface_hub==0.26.1
+image_reward==1.5
+mpi4py==4.0.1
+nirvana_dl==66.0.3
+numpy==2.1.2
+omegaconf==2.3.0
+pandas==2.2.3
+Pillow==11.0.0
+pytz==2024.2
+PyYAML==6.0.2
+PyYAML==6.0.2
+scipy==1.14.1
+tensorflow_io==0.37.1
+torch==2.5.0
+torchvision==0.20.0
+tqdm==4.66.5
+transformers==4.45.2
+typed_argument_parser==1.10.1
+wandb==0.18.5
+yt==4.3.1