Eyalgut commited on
Commit
28b1892
1 Parent(s): f6d485f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -88,13 +88,14 @@ pip install diffusers
88
  from diffusers import UNet2DConditionModel, DiffusionPipeline, LCMScheduler
89
  import torch
90
 
91
- unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.3-FAST-LORA", torch_dtype=torch.float16)
92
- pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3-BETA", unet=unet, torch_dtype=torch.float16)
 
93
 
94
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
95
  pipe.to("cuda")
96
 
97
  prompt = "A portrait of a Beautiful and playful ethereal singer, golden designs, highly detailed, blurry background"
98
 
99
- image = pipe(prompt, num_inference_steps=8, guidance_scale=5.0).images[0]
100
  ```
 
88
  from diffusers import UNet2DConditionModel, DiffusionPipeline, LCMScheduler
89
  import torch
90
 
91
+ pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3", unet=unet, torch_dtype=torch.float16)
92
+ pipeline.load_lora_weights("briaai/BRIA-2.3-FAST-LORA")
93
+ pipeline.fuse_lora()
94
 
95
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
96
  pipe.to("cuda")
97
 
98
  prompt = "A portrait of a Beautiful and playful ethereal singer, golden designs, highly detailed, blurry background"
99
 
100
+ image = pipe(prompt, num_inference_steps=8, guidance_scale=0.0).images[0]
101
  ```