Update models_server.py

models_server.py

@@ -170,8 +170,8 @@ def load_pipes(local_deployment):
             #     "model": DiffusionPipeline.from_pretrained(f"{local_models}lambdalabs/sd-image-variations-diffusers"), #torch_dtype=torch.float16
             #     "device": "cuda:0"
             # },
-            "bdsqlsz/stable-diffusion-v1-5": {
-                "model": DiffusionPipeline.from_pretrained(f"{local_models}bdsqlsz/stable-diffusion-v1-5"),
+            "CompVis/stable-diffusion-v1-4": {
+                "model": DiffusionPipeline.from_pretrained(f"{local_models}CompVis/stable-diffusion-v1-4"),
                 "device": "cuda:0"
             },
             # "superb/wav2vec2-base-superb-ks": {
@@ -268,7 +268,7 @@ def load_pipes(local_deployment):
         
         controlnet = ControlNetModel.from_pretrained(f"{local_models}lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
         controlnetpipe = StableDiffusionControlNetPipeline.from_pretrained(
-            f"{local_models}bdsqlsz/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
+            f"{local_models}CompVis/stable-diffusion-v1-4", controlnet=controlnet, torch_dtype=torch.float16
         )
 
 
@@ -463,7 +463,7 @@ def models(model_id, data):
             result = {"generated text": generated_text}
 
         # text to image
-        if model_id == "bdsqlsz/stable-diffusion-v1-5":
+        if model_id == "CompVis/stable-diffusion-v1-4":
             file_name = str(uuid.uuid4())[:4]
             text = data["text"]
             out = pipe(prompt=text)