jayparmr commited on
Commit
f025569
·
1 Parent(s): ecab1e0

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +6 -6
handler.py CHANGED
@@ -188,9 +188,9 @@ if device.type != 'cuda':
188
 
189
  # multi-model list
190
  multi_model_list = [
191
- {"model_id": "CompVis/stable-diffusion-v-1-4-original"},
192
- # {"model_id": "Bingsu/my-korean-stable-diffusion-v1-5"},
193
- # {"model_id": "jcplus/stable-diffusion-v1-5"},
194
  ]
195
 
196
  class EndpointHandler():
@@ -213,9 +213,9 @@ class EndpointHandler():
213
  # self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
214
  # self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(model["model_id"])
215
  # self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(model["model_id"])
216
- self.multi_controlnet_model[model["model_id"]] = controlnet.load(path + model["model_id"])
217
- self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(path + model["model_id"])
218
- self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(path + model["model_id"])
219
 
220
  print(" Logs: model[model_id]", model["model_id"])
221
  print("Logs: multimodel controlnet pipelines are", path + model["model_id"])
 
188
 
189
  # multi-model list
190
  multi_model_list = [
191
+ {"model_id": "jayparmr/icbinp"},
192
+ {"model_id": "Bingsu/my-korean-stable-diffusion-v1-5"},
193
+ {"model_id": "jcplus/stable-diffusion-v1-5"},
194
  ]
195
 
196
  class EndpointHandler():
 
213
  # self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
214
  # self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(model["model_id"])
215
  # self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(model["model_id"])
216
+ self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
217
+ self.multi_text2image_model[model["model_id"]] = text2img_pipe.load( model["model_id"])
218
+ self.multi_image2image_model[model["model_id"]] = img2img_pipe.load( model["model_id"])
219
 
220
  print(" Logs: model[model_id]", model["model_id"])
221
  print("Logs: multimodel controlnet pipelines are", path + model["model_id"])