Image-to-Image
Diffusers
Safetensors
FluxKontextPipeline
flux
kontext
tryon
diffusion
image-generation
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("mike-srp/flux-kontext-main", 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]flux-kontext-main
Flux-Kontext main model for virtual try-on applications
使用方法
这个模型是 gem-tryon-kontext 服务的一部分，用于虚拟试衣应用。
模型文件
flux1-kontext-dev.safetensors: 主要的扩散模型text_encoder/: 文本编码器text_encoder_2/: 第二个文本编码器vae/: VAE模型ae.safetensors: 自编码器- 各种配置文件和tokenizer
来源
这个模型来自 gem-tryon-kontext 项目，用于虚拟试衣应用。
许可证
Apache 2.0
- Downloads last month
- 8