tinyllava commited on
Commit
3915e08
1 Parent(s): 4990fff

Update modeling_tinyllava_phi.py

Browse files
Files changed (1) hide show
  1. modeling_tinyllava_phi.py +4 -4
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