update for requirements

eric-ai-lab · Dec 12, 2024 · 6356764 · 6356764
1 parent 2121c74
commit 6356764
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -22,7 +22,7 @@ Clone our repo and create a new python environment.
 ```bash
 git clone https://github.com/eric-ai-lab/MiniGPT-5.git
 cd MiniGPT-5
-conda create -n minigpt5 python=3.9
+conda create -n minigpt5 python=3.10
 conda activate minigpt5
 pip install -r requirements.txt
 ```

diff --git a/model.py b/model.py
@@ -19,7 +19,7 @@
 from utils import plot_images_and_text
 from constants import *
 from diffusers import StableDiffusionPipeline
-from diffusers.models.vae import DiagonalGaussianDistribution
+from diffusers.models.autoencoders.vae import DiagonalGaussianDistribution
 
 class MiniGPT5_InputProcessor(object):
 

diff --git a/requirements.txt b/requirements.txt
@@ -46,7 +46,7 @@ transformers==4.31.0
 # xformers
 accelerate>=0.20.3
 opencv-python
-peft
+peft==0.10.0
 tokenizers
 lightning>=2.0.2
 open_clip_torch

diff --git a/train_eval.py b/train_eval.py
@@ -18,7 +18,7 @@
 
 from lightning.pytorch.callbacks import BasePredictionWriter
 from model import MiniGPT5_Model, MiniGPT5_InputProcessor
-from metric import *
+# from metric import *
 
 class PredWriter(BasePredictionWriter):
     def write_on_epoch_end(