Skip to content
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.

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 clip model support #116

Merged
merged 8 commits into from
Oct 10, 2022
Merged

Open clip model support #116

merged 8 commits into from
Oct 10, 2022

Conversation

wanliAlex
Copy link
Collaborator

@wanliAlex wanliAlex commented Oct 7, 2022

  • Please check if the PR fulfills these requirements
  • The commit message follows our guidelines
  • Tests for the changes have been added (for bug fixes/features)
  • Docs have been added / updated (for bug fixes / features)
  • What kind of change does this PR introduce? (Bug fix, feature, docs update, ...)

  • Implementing the clip models from open_clip

  • https://github.com/mlfoundations/open_clip

  • What is the current behavior? (You can also link to an open issue here)

  • we only support clip models from openai

  • What is the new behavior (if this is a feature change)?

  • we can use clip models from open_clip

  • Does this PR introduce a breaking change? (What changes might users need to make in their application due to this PR?)

  • users need to install open_clip_torch package

  • Other information:

  • I update the requirements.txt and set_up.py files to support open_clip_torch.

@wanliAlex
Copy link
Collaborator Author

image

The changes passe the updated test.

src/marqo/s2_inference/clip_utils.py Outdated Show resolved Hide resolved
src/marqo/s2_inference/clip_utils.py Outdated Show resolved Hide resolved
src/marqo/s2_inference/model_registry.py Outdated Show resolved Hide resolved
tests/s2_inference/test_encoding.py Show resolved Hide resolved
@jn2clark jn2clark self-requested a review October 9, 2022 23:36
setup.py Outdated Show resolved Hide resolved
@wanliAlex wanliAlex merged commit 565f015 into mainline Oct 10, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants