Vision-CAIR commited on
Commit
9fb853f
1 Parent(s): 2ec8ca7

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. mini_gpt4_llama_v2.py +1 -0
mini_gpt4_llama_v2.py CHANGED
@@ -17,6 +17,7 @@ from torch.cuda.amp import autocast as autocast
17
  import torch.nn as nn
18
 
19
  from .registry import registry
 
20
  from .blip2 import Blip2Base, disabled_train
21
  from .conversation import Conversation, SeparatorStyle, StoppingCriteriaList, StoppingCriteriaSub
22
  from transformers import LlamaTokenizer
 
17
  import torch.nn as nn
18
 
19
  from .registry import registry
20
+ from .blip_processors import *
21
  from .blip2 import Blip2Base, disabled_train
22
  from .conversation import Conversation, SeparatorStyle, StoppingCriteriaList, StoppingCriteriaSub
23
  from transformers import LlamaTokenizer