tianweiy commited on
Commit
f72fab5
1 Parent(s): 19179ac

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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"