---
license: other
tags:
- text-to-image
- flux
- lora
- diffusers
- tryon
- virtual-try-on
- kontext
base_model: black-forest-labs/FLUX.1-dev
instance_prompt: a photo
widget:
- text: "a photo of person wearing elegant dress"
---

# Kontext Tryon Universal - 1031

这是一个基于 FLUX.1 的通用虚拟试穿（Virtual Try-On）LoRA 模型。

## 模型信息

- **基础模型**: FLUX.1-dev
- **训练方法**: LoRA (Low-Rank Adaptation)
- **训练数据**: Kontext 1022 merged with real (filtered)
- **Epoch**: 4
- **文件大小**: 4.6 GB
- **文件格式**: SafeTensors

## 使用方法

### 使用 Diffusers 库

```python
from diffusers import FluxPipeline
import torch

# 加载基础模型
pipe = FluxPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    torch_dtype=torch.bfloat16
)

# 加载 LoRA 权重
pipe.load_lora_weights(
    "mike-srp/kontext-tryon-universal-1031",
    weight_name="epoch-4.safetensors"
)

pipe.to("cuda")

# 生成图像
prompt = "a photo of person wearing elegant dress"
image = pipe(
    prompt,
    num_inference_steps=50,
    guidance_scale=7.5
).images[0]

image.save("tryon_result.png")
```

### 下载模型文件

```python
from huggingface_hub import hf_hub_download

model_path = hf_hub_download(
    repo_id="mike-srp/kontext-tryon-universal-1031",
    filename="epoch-4.safetensors"
)

print(f"模型已下载到: {model_path}")
```

### 在你的推理服务中使用

```python
import os
from diffusers import FluxPipeline

# 设置 LoRA 路径
lora_path = "mike-srp/kontext-tryon-universal-1031"

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
pipe.load_lora_weights(lora_path, weight_name="epoch-4.safetensors")
```

## 适用场景

- 虚拟试穿
- 服装展示
- 时尚设计
- 电商产品展示

## 许可证

请遵守 FLUX.1 的使用条款和相关许可证。

## 作者

- **Organization**: [srpone](https://huggingface.co/srpone)
- **Author**: [mike-srp](https://huggingface.co/mike-srp)