ibm-granite
/

granite-vision-3.3-2b-embedding

Feature Extraction

granitevisionemb

Model card Files Files and versions

Adirazgold commited on Jun 8

Commit

7de027c

·

verified ·

1 Parent(s): 2f5f4ea

Update modeling_colgranitevision.py

Files changed (1) hide show

modeling_colgranitevision.py +2 -2

modeling_colgranitevision.py CHANGED Viewed

@@ -5,8 +5,8 @@ from torch import nn
 from transformers import LlavaNextConfig, \
     LlavaNextPreTrainedModel
-from custom_llava_next import LlavaNextWithCustomPacking as LlavaNextForConditionalGeneration
-from colgranitevision_config import  ColGraniteVisionConfig
 # from transformers.models.paligemma.modeling_paligemma import (
 #     PaliGemmaConfig,

 from transformers import LlavaNextConfig, \
     LlavaNextPreTrainedModel
+from .custom_llava_next import LlavaNextWithCustomPacking as LlavaNextForConditionalGeneration
+from .colgranitevision_config import  ColGraniteVisionConfig
 # from transformers.models.paligemma.modeling_paligemma import (
 #     PaliGemmaConfig,