Vision-CAIR
commited on
Commit
•
9fb853f
1
Parent(s):
2ec8ca7
Upload folder using huggingface_hub
Browse files- 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
|