Skip to content

create importable visual bge #1387

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Diogenesoftoronto wants to merge 1 commit into
base: master
Choose a base branch
from
Open

create importable visual bge #1387

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file added research/__init__.py
View file
Open in desktop
Empty file.
1 change: 0 additions & 1 deletion research/visual_bge/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1 +0,0 @@
from .modeling import Visualized_BGE
2 changes: 1 addition & 1 deletion research/visual_bge/visual_bge/eva_clip/hf_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
2 changes: 1 addition & 1 deletion research/visual_bge/visual_bge/eva_clip/modified_resnet.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
4 changes: 3 additions & 1 deletion research/visual_bge/visual_bge/modeling.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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__)


Expand Down