Update handler.py
Browse files- handler.py +3 -1
handler.py
CHANGED
|
@@ -188,7 +188,9 @@ if device.type != 'cuda':
|
|
| 188 |
|
| 189 |
# multi-model list
|
| 190 |
multi_model_list = [
|
| 191 |
-
{"model_id": "/
|
|
|
|
|
|
|
| 192 |
]
|
| 193 |
|
| 194 |
class EndpointHandler():
|
|
|
|
| 188 |
|
| 189 |
# multi-model list
|
| 190 |
multi_model_list = [
|
| 191 |
+
{"model_id": "runwayml/stable-diffusion-v1-5"},
|
| 192 |
+
{"model_id": "Bingsu/my-korean-stable-diffusion-v1-5"},
|
| 193 |
+
{"model_id": "jcplus/stable-diffusion-v1-5"},
|
| 194 |
]
|
| 195 |
|
| 196 |
class EndpointHandler():
|