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
|