Update modeling_tinyllava_elm.py
Browse files
modeling_tinyllava_elm.py
CHANGED
|
@@ -20,7 +20,7 @@ from transformers.modeling_outputs import CausalLMOutputWithPast
|
|
| 20 |
from transformers.generation.utils import GenerateOutput
|
| 21 |
from transformers import AutoConfig, AutoModelForCausalLM, AutoModel, AutoImageProcessor
|
| 22 |
|
| 23 |
-
from configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
|
| 24 |
|
| 25 |
# from tinyllava.utils.data_utils import get_value_from_kwargs
|
| 26 |
CONTROLLER_HEART_BEAT_EXPIRATION = 30
|
|
|
|
| 20 |
from transformers.generation.utils import GenerateOutput
|
| 21 |
from transformers import AutoConfig, AutoModelForCausalLM, AutoModel, AutoImageProcessor
|
| 22 |
|
| 23 |
+
from .configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
|
| 24 |
|
| 25 |
# from tinyllava.utils.data_utils import get_value_from_kwargs
|
| 26 |
CONTROLLER_HEART_BEAT_EXPIRATION = 30
|