initial commit

models/common/vqa_tools/aokvqa/environment.yml

@@ -0,0 +1,36 @@
+name: aokvqa
+channels:
+  - pytorch
+  - nvidia
+  - huggingface
+  - conda-forge
+  - defaults
+dependencies:
+  - python=3.7
+  - cudatoolkit=11.3
+  - numpy=1.21.6
+  - pytorch=1.11.0
+  - torchvision=0.12.0
+  - pytorch-lightning=1.6.3
+  - torchmetrics=0.8.1
+  - gdown=4.4.0
+  - pip=22.0.4
+  - pip:
+    - argparse==1.4.0
+    - Pillow==9.0.1
+    - tensorboard==2.9.0
+    - ftfy==6.1.1
+    - regex==2022.3.15
+    - tqdm==4.64.0
+    - clip @ git+https://github.com/openai/CLIP.git@b46f5ac7587d2e1862f8b7b1573179d80dcdd620
+    - openai==0.18.1
+    - nltk==3.7
+    - sacrebleu==2.0.0
+    - sacremoses==0.0.53
+    - sentence-transformers==2.2.0
+    - datasets==2.1.0
+    - tokenizers==0.10.3
+    - transformers==4.10.3
+
+# Next: resolve conflict between sentence-transfomers and pytorch-lightning
+# pip uninstall sentencepiece