Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ ckpt_name = "dmd2_sdxl_4step_lora_fp16.safetensors"
|
|
64 |
# Load model.
|
65 |
pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
66 |
pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
|
67 |
-
pipe.fuse_lora()
|
68 |
|
69 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
70 |
prompt="a photo of a cat"
|
|
|
64 |
# Load model.
|
65 |
pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
66 |
pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
|
67 |
+
pipe.fuse_lora(lora_scale=1.0) # we might want to make the scale smaller for community models
|
68 |
|
69 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
70 |
prompt="a photo of a cat"
|