diff --git a/research/__init__.py b/research/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/research/visual_bge/__init__.py b/research/visual_bge/__init__.py index 133aef26..e69de29b 100644 --- a/research/visual_bge/__init__.py +++ b/research/visual_bge/__init__.py @@ -1 +0,0 @@ -from .modeling import Visualized_BGE \ No newline at end of file diff --git a/research/visual_bge/visual_bge/eva_clip/hf_model.py b/research/visual_bge/visual_bge/eva_clip/hf_model.py index c4b9fd85..bbbd6be8 100644 --- a/research/visual_bge/visual_bge/eva_clip/hf_model.py +++ b/research/visual_bge/visual_bge/eva_clip/hf_model.py @@ -25,7 +25,7 @@ class BaseModelOutput: class PretrainedConfig: pass -from .hf_configs import arch_dict +from FlagEmbedding.research.visual_bge.visual_bge.eva_clip.hf_configs import arch_dict # utils def _camel2snake(s): diff --git a/research/visual_bge/visual_bge/eva_clip/modified_resnet.py b/research/visual_bge/visual_bge/eva_clip/modified_resnet.py index 3be17ce9..6a8d3aed 100644 --- a/research/visual_bge/visual_bge/eva_clip/modified_resnet.py +++ b/research/visual_bge/visual_bge/eva_clip/modified_resnet.py @@ -4,7 +4,7 @@ from torch import nn from torch.nn import functional as F -from visual_bge.eva_clip.utils import freeze_batch_norm_2d +from .utils import freeze_batch_norm_2d class Bottleneck(nn.Module): diff --git a/research/visual_bge/visual_bge/modeling.py b/research/visual_bge/visual_bge/modeling.py index deecada4..bccc116e 100644 --- a/research/visual_bge/visual_bge/modeling.py +++ b/research/visual_bge/visual_bge/modeling.py @@ -7,11 +7,13 @@ from torch import nn, Tensor from transformers import AutoModel, AutoTokenizer, AutoConfig from transformers.file_utils import ModelOutput +from PIL import ImageFile -from visual_bge.eva_clip import create_eva_vision_and_transforms +from .eva_clip import create_eva_vision_and_transforms from PIL import Image +ImageFile.LOAD_TRUNCATED_IMAGES = True logger = logging.getLogger(__name__)