VikramSingh178
commited on
Commit
•
1bb3281
1
Parent(s):
b172fe6
Update README.md
Browse files
README.md
CHANGED
@@ -1,16 +1,37 @@
|
|
1 |
---
|
2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
datasets:
|
4 |
- hahminlew/kream-product-blip-captions
|
5 |
language:
|
6 |
- en
|
7 |
-
library_name: diffusers
|
8 |
pipeline_tag: text-to-image
|
9 |
---
|
10 |
-
---
|
11 |
-
license: mit
|
12 |
-
language:
|
13 |
-
- en
|
14 |
|
15 |
-
|
16 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
license: apache-2.0
|
3 |
+
base_model: stabilityai/stable-diffusion-xl-base-1.0
|
4 |
+
dataset: hahminlew/kream-product-blip-captions
|
5 |
+
tags:
|
6 |
+
- stable-diffusion-xl
|
7 |
+
- stable-diffusion-xl-diffusers
|
8 |
+
- text-to-image
|
9 |
+
- diffusers
|
10 |
+
- lora
|
11 |
+
- stable-diffusion
|
12 |
+
- stable-diffusion-diffusers
|
13 |
+
inference: true
|
14 |
datasets:
|
15 |
- hahminlew/kream-product-blip-captions
|
16 |
language:
|
17 |
- en
|
|
|
18 |
pipeline_tag: text-to-image
|
19 |
---
|
|
|
|
|
|
|
|
|
20 |
|
21 |
+
|
22 |
+
# Inference
|
23 |
+
|
24 |
+
```python
|
25 |
+
from diffusers import DiffusionPipeline
|
26 |
+
import torch
|
27 |
+
|
28 |
+
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
29 |
+
pipe.to("cuda")
|
30 |
+
pipe.load_lora_weights("VikramSingh178/sdxl-lora-finetune-product-caption")
|
31 |
+
|
32 |
+
prompt = "outer, The Nike x Balenciaga Down Jacket Black, a photography of a black down jacket with a logo on the chest."
|
33 |
+
|
34 |
+
image = pipe(prompt, num_inference_steps=45, guidance_scale=7.5).images[0]
|
35 |
+
image.save("example.png")
|
36 |
+
```
|
37 |
+
|