Upload folder using huggingface_hub

mini_gpt4_llama_v2.py

@@ -102,7 +102,8 @@ class MiniGPT4_Video(Blip2Base, PreTrainedModel):
         Blip2Base.__init__(self)
         
         vis_processor_cfg = {"name": "blip2_image_train","image_size": 224}
-        self.vis_processor = registry.get_processor_class(vis_processor_cfg["name"]).from_config(vis_processor_cfg)
+        self.vis_processor = registry.get_processor_class(vis_processor_cfg["name"])
+        self.vis_processor = self.vis_processor.from_config(vis_processor_cfg)
         self.CONV_VISION = CONV_VISION
         if "Mistral" in self.llama_model:
             from .modeling_mistral import MistralForCausalLM as llm_model

registry.py

@@ -243,6 +243,7 @@ class Registry:
 
     @classmethod
     def get_processor_class(cls, name):
+        print(cls.mapping["processor_name_mapping"])
         return cls.mapping["processor_name_mapping"].get(name, None)
 
     @classmethod

utils.py

@@ -23,7 +23,7 @@ import pandas as pd
 import yaml
 from iopath.common.download import download
 from iopath.common.file_io import file_lock, g_pathmgr
-from .registry import registry
+from minigpt4_video.registry import registry
 from torch.utils.model_zoo import tqdm
 from torchvision.datasets.utils import (
     check_integrity,