From 9b40fa5a29d17b106d588054ac92d1f76dd501c4 Mon Sep 17 00:00:00 2001
From: Diogenesoftoronto <87236699+Diogenesoftoronto@users.noreply.github.com>
Date: Thu, 27 Feb 2025 08:23:09 -0500
Subject: [PATCH] create importable visual bge

---
 research/__init__.py                                       | 0
 research/visual_bge/__init__.py                            | 1 -
 research/visual_bge/visual_bge/eva_clip/hf_model.py        | 2 +-
 research/visual_bge/visual_bge/eva_clip/modified_resnet.py | 2 +-
 research/visual_bge/visual_bge/modeling.py                 | 4 +++-
 5 files changed, 5 insertions(+), 4 deletions(-)
 create mode 100644 research/__init__.py

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__)