README.md

---
tags:
- text-to-image
- style-transfer
- stable-diffusion
language:
- en
library_name: diffusers
license: mit
---

# StyleShot Model Card


<div align="center">

[**Project Page**](https://styleshot.github.io) **|** [**Paper (ArXiv)**](https://arxiv.org/abs/2407.01414) **|** [**Code**](https://github.com/open-mmlab/StyleShot)
</div>

---


## Introduction

we present StyleShot, a generalized plug-to-play style transfer method, capable of generating the high-quality stylized images that match the desired style from any reference image without test-time style-tuning. To the best of our knowledge, To the best of our knowledge, StyleShot is the first work to designate a style-aware encoder based on Stable Diffusion and a content-fusion encoder for better style and content integration, achieving the state-of-the-art text and image-driven style transfer performance compared to existing methods.

![arch](./framework.png)

## Models

### StyleShot for SD 1.5
- [ip.bin](https://huggingface.co/Gaojunyao/StyleShot/blob/main/pretrained_weight/ip.bin): weights for ip-adapter and our content-retention encoder.
- [style_aware_encoder.bin](https://huggingface.co/Gaojunyao/StyleShot/blob/main/pretrained_weight/style_aware_encoder.bin): weights for style-aware encoder.

## Disclaimer
We develop this repository for RESEARCH purposes, so it can only be used for personal/research/non-commercial purposes.