jayparmr
/

model-sd-multi

Model card Files Files and versions

jayparmr commited on May 24, 2023

Commit

ecab1e0

·

1 Parent(s): e9b6262

Update handler.py

Files changed (1) hide show

handler.py +3 -3

handler.py CHANGED Viewed

@@ -188,9 +188,9 @@ if device.type != 'cuda':
 # multi-model list
 multi_model_list = [
-    {"model_id": "runwayml/stable-diffusion-v1-5"},
-    {"model_id": "Bingsu/my-korean-stable-diffusion-v1-5"},
-    {"model_id": "jcplus/stable-diffusion-v1-5"},
 ]
 class EndpointHandler():

 # multi-model list
 multi_model_list = [
+    {"model_id": "CompVis/stable-diffusion-v-1-4-original"},
+    # {"model_id": "Bingsu/my-korean-stable-diffusion-v1-5"},
+    # {"model_id": "jcplus/stable-diffusion-v1-5"},
 ]
 class EndpointHandler():