Pytorch API to work with CLIP models.
Project description
pytorch_clip: Pytorch API to work with CLIP models.
Install package
pip install pytorch_clip
Install the latest version
pip install --upgrade git+https://github.com/bes-dev/pytorch_clip.git
Features
- Supports original CLIP models by OpenAI and ruCLIP model by SberAI.
Usage
Simple code
import torch
from pytorch_clip import get_models_list, get_clip_model
print(get_models_list())
model, text_processor, image_processor = get_clip_model()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pytorch_clip-2021.12.7.2.tar.gz
(10.1 kB
view hashes)