Update modeling_tinyllava_phi.py
Browse files
modeling_tinyllava_phi.py
CHANGED
@@ -1,8 +1,5 @@
|
|
1 |
|
2 |
from .configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
|
3 |
-
import sys
|
4 |
-
sys.path.append(os.path.dirname(sys.path[0]))
|
5 |
-
from . import test
|
6 |
|
7 |
#from .data_preprocess import load_image, process_images, tokenizer_image_token
|
8 |
from dataclasses import dataclass
|
@@ -22,7 +19,10 @@ from transformers import CLIPVisionModel, CLIPImageProcessor, SiglipVisionModel,
|
|
22 |
|
23 |
import time
|
24 |
from transformers import AutoConfig, AutoModelForCausalLM, PhiForCausalLM
|
25 |
-
|
|
|
|
|
|
|
26 |
|
27 |
# from tinyllava.utils.data_utils import get_value_from_kwargs
|
28 |
CONTROLLER_HEART_BEAT_EXPIRATION = 30
|
|
|
1 |
|
2 |
from .configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
|
|
|
|
|
|
|
3 |
|
4 |
#from .data_preprocess import load_image, process_images, tokenizer_image_token
|
5 |
from dataclasses import dataclass
|
|
|
19 |
|
20 |
import time
|
21 |
from transformers import AutoConfig, AutoModelForCausalLM, PhiForCausalLM
|
22 |
+
import os
|
23 |
+
import sys
|
24 |
+
sys.path.append(os.path.dirname(sys.path[0]))
|
25 |
+
from . import test
|
26 |
|
27 |
# from tinyllava.utils.data_utils import get_value_from_kwargs
|
28 |
CONTROLLER_HEART_BEAT_EXPIRATION = 30
|