chore: m2 installable

environment.yml

@@ -3,12 +3,11 @@ channels:
   - pytorch
   - defaults
   - anaconda
+  - conda-forge
 dependencies:
   - python=3.9
-  - cudatoolkit
   - pip
   - pytorch=1.12.1
-  - pytorch-mutex=1.0=cuda
   - torchaudio=0.12.1
   - torchvision=0.13.1
   - pip:
@@ -30,6 +29,7 @@ dependencies:
     - kiwisolver==1.4.4
     - matplotlib==3.7.0
     - multidict==6.0.4
+    - numpy
     - openai==0.27.0
     - packaging==23.0
     - psutil==5.9.4
@@ -51,7 +51,6 @@ dependencies:
     - omegaconf==2.3.0
     - opencv-python==4.7.0.72
     - iopath==0.1.10
-    - decord==0.6.0
     - tenacity==8.2.2
     - peft
     - pycocoevalcap

minigpt4/datasets/data_utils.py

@@ -5,20 +5,12 @@
  For full license text, see the LICENSE_Lavis file in the repo root or https://opensource.org/licenses/BSD-3-Clause
 """
 
-import gzip
 import logging
-import os
-import random as rnd
-import tarfile
-import zipfile
 import random
 from typing import List
-from tqdm import tqdm
 
-import decord
-from decord import VideoReader
+# import decord
 import webdataset as wds
-import numpy as np
 import torch
 from torch.utils.data.dataset import IterableDataset
 
@@ -26,7 +18,7 @@ from minigpt4.common.registry import registry
 from minigpt4.datasets.datasets.base_dataset import ConcatDataset
 
 
-decord.bridge.set_bridge("torch")
+# decord.bridge.set_bridge("torch")
 MAX_INT = registry.get("MAX_INT")