Update README.md
Browse files
README.md
CHANGED
@@ -89,8 +89,8 @@ from diffusers import DiffusionPipeline, LCMScheduler
|
|
89 |
import torch
|
90 |
|
91 |
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.float16)
|
92 |
-
|
93 |
-
|
94 |
|
95 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
96 |
pipe.to("cuda")
|
|
|
89 |
import torch
|
90 |
|
91 |
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.float16)
|
92 |
+
pipe.load_lora_weights("briaai/BRIA-2.3-FAST-LORA")
|
93 |
+
pipe.fuse_lora()
|
94 |
|
95 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
96 |
pipe.to("cuda")
|