You can deploy this model and start earning money today!

HZSDU/
Exploration_Platform

Views

No views yet

Attention Distillation: A Unified Approach to Visual Characteristics Transfer

2025/03/08: We provide a new notebook with Style-specific T2I Generation with Flux.1-dev. See Issue 1 for more details.
2025/03/05: We add tiling to enable seamless textures generation. See Issue 3 for more details.
2025/03/01: We provide a simple HuggingFace🤗 demo. Check it out here !

This code was tested with Python 3.10, Pytorch 2.5 and Diffusers 0.32.

See [Texture Synthesis] part of ad notebook for generating texture images using SD1.5.

See [Style/Appearance Transfer] part of ad notebook for style/appearance transfer using SD1.5.

See [Style-specific T2I Generation] part of ad notebook for style-specific T2I generation using SD1.5 or SDXL.

bash

python train_vae.py \
    --image_path=/path/to/image \
    --vae_model_path=/path/to/vae

Run the following command to start the Web UI:

python app.py

The Web UI will be available at http://localhost:7860.

We also provide an implementation of Attention Distillation for ComfyUI. For more details, see here.