Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ You can use the pipeline like so:
|
|
31 |
from diffusers import DiffusionPipeline
|
32 |
import torch
|
33 |
|
34 |
-
pipeline = DiffusionPipeline.from_pretrained("
|
35 |
#Load LoRA finetune
|
36 |
pipeline.load_lora_weights("segmind/tiny_lora_mxtun3_style", weight_name="sd15_tiny_mxtun3_style_lora.safetensors")
|
37 |
prompt = "Portrait of a pretty girl"
|
|
|
31 |
from diffusers import DiffusionPipeline
|
32 |
import torch
|
33 |
|
34 |
+
pipeline = DiffusionPipeline.from_pretrained("segmind/tiny-sd", torch_dtype=torch.float16)
|
35 |
#Load LoRA finetune
|
36 |
pipeline.load_lora_weights("segmind/tiny_lora_mxtun3_style", weight_name="sd15_tiny_mxtun3_style_lora.safetensors")
|
37 |
prompt = "Portrait of a pretty girl"
|