update readme and environment.yml

README.md

@@ -1,10 +1,16 @@
 # MiniGPT-4 and MiniGPT-v2
 
+**MiniGPT-v2: Large Language Model as a Unified Interface for Vision-Language Multi-task Learning**
 
-**King Abdullah University of Science and Technology**
+<a href='https://minigpt-v2.github.io'><img src='https://img.shields.io/badge/Project-Page-Green'></a>   
+
+
+**MiniGPT-4: Enhancing Vision-language Understanding with Advanced Large Language Models**
 
 <a href='https://minigpt-4.github.io'><img src='https://img.shields.io/badge/Project-Page-Green'></a>  <a href='https://arxiv.org/abs/2304.10592'><img src='https://img.shields.io/badge/Paper-Arxiv-red'></a> <a href='https://huggingface.co/spaces/Vision-CAIR/minigpt4'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue'></a> <a href='https://huggingface.co/Vision-CAIR/MiniGPT-4'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Model-blue'></a> [![Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1OK4kYsZphwt5DXchKkzMBjYF6jnkqh4R?usp=sharing) [![YouTube](https://badges.aleen42.com/src/youtube.svg)](https://www.youtube.com/watch?v=__tftoxpBAw&feature=youtu.be)
 
+*King Abdullah University of Science and Technology*
+
 ## 💡 Get help - [Q&A](https://github.com/Vision-CAIR/MiniGPT-4/discussions/categories/q-a) or [Discord 💬](https://discord.gg/5WdJkjbAeE)
 
 

environment.yml

@@ -7,56 +7,27 @@ dependencies:
   - python=3.9
   - cudatoolkit
   - pip
-  - pytorch=2.0.0
-  - pytorch-mutex=1.0=cuda
-  - torchaudio=0.12.1
-  - torchvision=0.13.1
   - pip:
-    - accelerate==0.20.3
-    - aiohttp==3.8.4
-    - aiosignal==1.3.1
-    - async-timeout==4.0.2
-    - attrs==22.2.0
-    - bitsandbytes==0.37.0
-    - cchardet==2.1.7
-    - chardet==5.1.0
-    - contourpy==1.0.7
-    - cycler==0.11.0
-    - filelock==3.9.0
-    - fonttools==4.38.0
-    - frozenlist==1.3.3
+    - torch==2.0.0
+    - torchaudio
+    - torchvision
     - huggingface-hub==0.18.0
-    - importlib-resources==5.12.0
-    - kiwisolver==1.4.4
     - matplotlib==3.7.0
-    - multidict==6.0.4
-    - openai==0.27.0
-    - packaging==23.0
     - psutil==5.9.4
-    - pycocotools==2.0.6
-    - pyparsing==3.0.9
-    - python-dateutil==2.8.2
+    - iopath
     - pyyaml==6.0
     - regex==2022.10.31
     - tokenizers==0.13.2
     - tqdm==4.64.1
-    - transformers==4.32.0
+    - transformers==4.30.0
     - timm==0.6.13
-    - spacy==3.5.1
     - webdataset==0.2.48
-    - scikit-learn==1.2.2
-    - scipy==1.10.1
-    - yarl==1.8.2
-    - zipp==3.14.0
     - omegaconf==2.3.0
     - opencv-python==4.7.0.72
-    - iopath==0.1.10
     - decord==0.6.0
-    - tenacity==8.2.2
     - peft==0.2.0
-    - pycocoevalcap
     - sentence-transformers
-    - umap-learn
-    - notebook
     - gradio==3.47.1
+    - accelerate==0.20.3
+    - bitsandbytes==0.37.0
     - wandb