---
license: mit
datasets:
- tomg-group-umd/ContraStyles
library_name: transformers
---
An unofficial implementation of [CSD](https://github.com/learn2phoenix/CSD)

Inspired by [vvmatorin/CSD](https://huggingface.co./vvmatorin/CSD), the difference in this implementation is that the CLIP backbone is not the openai clip class but an instance of `CLIPVisionModel`.

Inference:

```python
from PIL import Image
from transformers import AutoProcessor, AutoModel

model = AutoModel.from_pretrained("NagaSaiAbhinay/CSD", trust_remote_code=True)
processor = AutoProcessor.from_pretrained("NagaSaiAbhinay/CSD")

img = Image.open('test_image.png')
pixel_values = processor

```