cpu4dream
/

llava-small-OpenELM-AIMv2-0.6B-auto

Image-Text-to-Text

Model card Files Files and versions Community

zcamz commited on Dec 18, 2024

Commit

300414f

·

verified ·

1 Parent(s): ec9267c

Update modeling_tinyllava_elm.py

Files changed (1) hide show

modeling_tinyllava_elm.py +1 -1

modeling_tinyllava_elm.py CHANGED Viewed

@@ -20,7 +20,7 @@ from transformers.modeling_outputs import CausalLMOutputWithPast
 from transformers.generation.utils import GenerateOutput
 from transformers import AutoConfig, AutoModelForCausalLM, AutoModel, AutoImageProcessor
-from configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
 # from tinyllava.utils.data_utils import get_value_from_kwargs
 CONTROLLER_HEART_BEAT_EXPIRATION = 30

 from transformers.generation.utils import GenerateOutput
 from transformers import AutoConfig, AutoModelForCausalLM, AutoModel, AutoImageProcessor
+from .configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
 # from tinyllava.utils.data_utils import get_value_from_kwargs
 CONTROLLER_HEART_BEAT_EXPIRATION = 30