Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 library_name: diffusers
-base_model: runwayml/stable-diffusion-v1-5
+base_model: segmind/SSD-1B
 tags:
 - lora
 - text-to-image
@@ -13,7 +13,7 @@ inference: false
 Latent Consistency Model (LCM) LoRA was proposed in [LCM-LoRA: A universal Stable-Diffusion Acceleration Module](TODO:) 
 by *Simian Luo, Yiqin Tan, Suraj Patil, Daniel Gu et al.*
 
-It is a distilled consistency adapter for [`segmind/SSD-1B`](https://huggingface.co/runwayml/stable-diffusion-v1-5) that allows
+It is a distilled consistency adapter for [`segmind/SSD-1B`](https://huggingface.co/segmind/SSD-1B) that allows
 to reduce the number of inference steps to only between **2 - 8 steps**.
 
 | Model                                                                      | Params / M | 
@@ -43,7 +43,7 @@ import torch
 from diffusers import LCMScheduler, AutoPipelineForText2Image
 
 model_id = "segmind/SSD-1B"
-adapter_id = "latent-consistency/lcm-lora-sdv1-5"
+adapter_id = "latent-consistency/lcm-lora-ssd-1b"
 
 pipe = AutoPipelineForText2Image.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16")
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)