Text-to-Image
Diffusers
English
lora
SkyAsl commited on
Commit
c70c9e0
·
verified ·
1 Parent(s): fd51d00

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -15,9 +15,9 @@ tags:
15
  ---
16
  ## Model description
17
 
18
- # sdxl-face-LoRA
19
 
20
- **sdxl-face-lora** is a LoRA fine-tuned adapter for the [stable-diffusion-xl-base-1.0](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0) model, trained on the [Rapidata/Face_Generation_Benchmark dataset](https://huggingface.co/datasets/Rapidata/Face_Generation_Benchmark). It specializes in generating face images.
21
 
22
  ---
23
 
@@ -29,17 +29,20 @@ tags:
29
  pip install diffusers transformers accelerate safetensor torch
30
  ```
31
 
32
- 2. **Load base Qwen-Image and LoRA adapter**:
33
 
34
  ```python
35
  from diffusers import DiffusionPipeline
36
  import torch
 
37
  pipe = DiffusionPipeline.from_pretrained(
38
  "stabilityai/stable-diffusion-xl-base-1.0",
39
  torch_dtype=torch.float16,
40
  ).to("cuda")
 
41
  pipe.load_lora_weights("SkyAsl/sdxl-face-lora")
42
  pipe.fuse_lora()
 
43
  image = pipe("a high-quality portrait of a woman").images[0]
44
  image.save("test.png")
45
  ```
 
15
  ---
16
  ## Model description
17
 
18
+ # Portrait-generator
19
 
20
+ **Portrait-generator** is a LoRA fine-tuned adapter for the [stable-diffusion-xl-base-1.0](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0) model, trained on the [Rapidata/Face_Generation_Benchmark dataset](https://huggingface.co/datasets/Rapidata/Face_Generation_Benchmark). It specializes in generating face images.
21
 
22
  ---
23
 
 
29
  pip install diffusers transformers accelerate safetensor torch
30
  ```
31
 
32
+ 2. **Load stable-diffusion-xl-base-1.0 and LoRA adapter**:
33
 
34
  ```python
35
  from diffusers import DiffusionPipeline
36
  import torch
37
+
38
  pipe = DiffusionPipeline.from_pretrained(
39
  "stabilityai/stable-diffusion-xl-base-1.0",
40
  torch_dtype=torch.float16,
41
  ).to("cuda")
42
+
43
  pipe.load_lora_weights("SkyAsl/sdxl-face-lora")
44
  pipe.fuse_lora()
45
+
46
  image = pipe("a high-quality portrait of a woman").images[0]
47
  image.save("test.png")
48
  ```