jayparmr commited on
Commit
1625be6
·
1 Parent(s): 219463f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -4
handler.py CHANGED
@@ -189,7 +189,7 @@ if device.type != 'cuda':
189
  multi_model_list = [
190
  {"model_id": "jayparmr/icbinp"},
191
  {"model_id": "jayparmr/comic_anime"},
192
- {"model_id": "jcplus/stable-diffusion-v1-5"},
193
  ]
194
 
195
  multi_controlnet_model={}
@@ -211,9 +211,7 @@ class EndpointHandler():
211
  for model in multi_model_list:
212
  print("Logs: model value is", model)
213
  print("Logs: model path value is", model["model_id"] )
214
- # self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
215
- # self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(model["model_id"])
216
- # self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(model["model_id"])
217
  controlnet = ControlNet()
218
  img2img_pipe = Img2Img()
219
  text2img_pipe = Text2Img()
 
189
  multi_model_list = [
190
  {"model_id": "jayparmr/icbinp"},
191
  {"model_id": "jayparmr/comic_anime"},
192
+ {"model_id": "jayparmr/icbinp"},
193
  ]
194
 
195
  multi_controlnet_model={}
 
211
  for model in multi_model_list:
212
  print("Logs: model value is", model)
213
  print("Logs: model path value is", model["model_id"] )
214
+
 
 
215
  controlnet = ControlNet()
216
  img2img_pipe = Img2Img()
217
  text2img_pipe = Text2Img()