Image-to-Image
Diffusers
How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Kunbyte/DRA-Ctrl", dtype=torch.bfloat16, device_map="cuda")

prompt = "Turn this cat into a dog"
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")

image = pipe(image=input_image, prompt=prompt).images[0]

Model Card for DRA-Ctrl

arXiv arXiv HuggingFace HuggingFace GitHub Project

This repository contains the LoRA weights for DRA-Ctrl across 9 tasks. For instructions on how to use these weights, please refer to our GitHub repository and HuggingFace Space.

Downloads last month
226
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Kunbyte/DRA-Ctrl

Finetuned
(33)
this model

Space using Kunbyte/DRA-Ctrl 1

Paper for Kunbyte/DRA-Ctrl