Commit
·
d3d1b30
1
Parent(s):
0d2b8bf
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,4 +2,24 @@
|
|
| 2 |
license: creativeml-openrail-m
|
| 3 |
library_name: diffusers
|
| 4 |
pipeline_tag: text-to-image
|
| 5 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: creativeml-openrail-m
|
| 3 |
library_name: diffusers
|
| 4 |
pipeline_tag: text-to-image
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
```
|
| 8 |
+
pipeline = DiffusionPipeline.from_pretrained(
|
| 9 |
+
"Andyrasika/lora_diffusion"
|
| 10 |
+
,custom_pipeline = "lpw_stable_diffusion"
|
| 11 |
+
,torch_dtype = torch.float16
|
| 12 |
+
)
|
| 13 |
+
lora = ("/content/lora_model.safetensors",0.8)
|
| 14 |
+
pipeline = __load_lora(pipeline=pipeline,lora_path=lora[0],lora_weight=lora[1])
|
| 15 |
+
pipeline.to("cuda")
|
| 16 |
+
# pipeline.enable_xformers_memory_efficient_attention()
|
| 17 |
+
#https://huggingface.co/docs/diffusers/optimization/fp16
|
| 18 |
+
pipeline.enable_vae_tiling()
|
| 19 |
+
prompt = """
|
| 20 |
+
shukezouma,negative space,shuimobysim
|
| 21 |
+
a branch of flower, traditional chinese ink painting
|
| 22 |
+
"""
|
| 23 |
+
image = pipeline(prompt).images[0]
|
| 24 |
+
image
|
| 25 |
+
```
|