mike-srp commited on
Commit
e31ee35
·
verified ·
1 Parent(s): 4993278

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +103 -0
README.md ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ tags:
4
+ - text-to-image
5
+ - flux
6
+ - lora
7
+ - diffusers
8
+ - tryon
9
+ - virtual-try-on
10
+ - kontext
11
+ base_model: black-forest-labs/FLUX.1-dev
12
+ instance_prompt: a photo
13
+ widget:
14
+ - text: "a photo of person wearing elegant dress"
15
+ ---
16
+
17
+ # Kontext Tryon Universal - 1031
18
+
19
+ 这是一个基于 FLUX.1 的通用虚拟试穿(Virtual Try-On)LoRA 模型。
20
+
21
+ ## 模型信息
22
+
23
+ - **基础模型**: FLUX.1-dev
24
+ - **训练方法**: LoRA (Low-Rank Adaptation)
25
+ - **训练数据**: Kontext 1022 merged with real (filtered)
26
+ - **Epoch**: 4
27
+ - **文件大小**: 4.6 GB
28
+ - **文件格式**: SafeTensors
29
+
30
+ ## 使用方法
31
+
32
+ ### 使用 Diffusers 库
33
+
34
+ ```python
35
+ from diffusers import FluxPipeline
36
+ import torch
37
+
38
+ # 加载基础模型
39
+ pipe = FluxPipeline.from_pretrained(
40
+ "black-forest-labs/FLUX.1-dev",
41
+ torch_dtype=torch.bfloat16
42
+ )
43
+
44
+ # 加载 LoRA 权重
45
+ pipe.load_lora_weights(
46
+ "mike-srp/kontext-tryon-universal-1031",
47
+ weight_name="epoch-4.safetensors"
48
+ )
49
+
50
+ pipe.to("cuda")
51
+
52
+ # 生成图像
53
+ prompt = "a photo of person wearing elegant dress"
54
+ image = pipe(
55
+ prompt,
56
+ num_inference_steps=50,
57
+ guidance_scale=7.5
58
+ ).images[0]
59
+
60
+ image.save("tryon_result.png")
61
+ ```
62
+
63
+ ### 下载模型文件
64
+
65
+ ```python
66
+ from huggingface_hub import hf_hub_download
67
+
68
+ model_path = hf_hub_download(
69
+ repo_id="mike-srp/kontext-tryon-universal-1031",
70
+ filename="epoch-4.safetensors"
71
+ )
72
+
73
+ print(f"模型已下载到: {model_path}")
74
+ ```
75
+
76
+ ### 在你的推理服务中使用
77
+
78
+ ```python
79
+ import os
80
+ from diffusers import FluxPipeline
81
+
82
+ # 设置 LoRA 路径
83
+ lora_path = "mike-srp/kontext-tryon-universal-1031"
84
+
85
+ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
86
+ pipe.load_lora_weights(lora_path, weight_name="epoch-4.safetensors")
87
+ ```
88
+
89
+ ## 适用场景
90
+
91
+ - 虚拟试穿
92
+ - 服装展示
93
+ - 时尚设计
94
+ - 电商产品展示
95
+
96
+ ## 许可证
97
+
98
+ 请遵守 FLUX.1 的使用条款和相关许可证。
99
+
100
+ ## 作者
101
+
102
+ - **Organization**: [srpone](https://huggingface.co/srpone)
103
+ - **Author**: [mike-srp](https://huggingface.co/mike-srp)