Update README.md
Browse files
README.md
CHANGED
@@ -85,10 +85,10 @@ pip install diffusers
|
|
85 |
|
86 |
|
87 |
```py
|
88 |
-
from diffusers import
|
89 |
import torch
|
90 |
|
91 |
-
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3",
|
92 |
pipeline.load_lora_weights("briaai/BRIA-2.3-FAST-LORA")
|
93 |
pipeline.fuse_lora()
|
94 |
|
|
|
85 |
|
86 |
|
87 |
```py
|
88 |
+
from diffusers import DiffusionPipeline, LCMScheduler
|
89 |
import torch
|
90 |
|
91 |
+
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.float16)
|
92 |
pipeline.load_lora_weights("briaai/BRIA-2.3-FAST-LORA")
|
93 |
pipeline.fuse_lora()
|
94 |
|