Skip to content

Commit

Permalink
remove unused torchvision dependency (#1881)
Browse files Browse the repository at this point in the history
  • Loading branch information
dvruette authored Dec 12, 2023
1 parent 944cac0 commit aba19ec
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 3 deletions.
2 changes: 1 addition & 1 deletion examples/applications/image-search/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ SentenceTransformers provides models that allow to embed images and text into th


## Installation
Ensure that you have [torchvision](https://pypi.org/project/torchvision/) installed to use the image-text-models and use a recent PyTorch version (tested with PyTorch 1.7.0). Image-Text-Models have been added with SentenceTransformers version 1.0.0. Image-Text-Models are still in an experimental phase.
Ensure that you have [transformers](https://pypi.org/project/transformers/) installed to use the image-text-models and use a recent PyTorch version (tested with PyTorch 1.7.0). Image-Text-Models have been added with SentenceTransformers version 1.0.0. Image-Text-Models are still in an experimental phase.

## Usage
SentenceTransformers provides a wrapper for the [OpenAI CLIP Model](https://github.com/openai/CLIP), which was trained on a variety of (image, text)-pairs.
Expand Down
1 change: 0 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@ transformers>=4.6.0,<5.0.0
tokenizers>=0.10.3
tqdm
torch>=1.6.0
torchvision
numpy
scikit-learn
scipy
Expand Down
1 change: 0 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@
'transformers>=4.6.0,<5.0.0',
'tqdm',
'torch>=1.6.0',
'torchvision',
'numpy',
'scikit-learn',
'scipy',
Expand Down

0 comments on commit aba19ec

Please sign in to comment.