Spaces:
Runtime error
Runtime error
Mehdi Cherti
commited on
Commit
·
a45817e
1
Parent(s):
7ac3881
update models
Browse files
run.py
CHANGED
|
@@ -236,6 +236,10 @@ def ddgan_laion2b_v2():
|
|
| 236 |
cfg = ddgan_sd_v9()
|
| 237 |
return cfg
|
| 238 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 239 |
def ddgan_sd_v11():
|
| 240 |
cfg = ddgan_sd_v10()
|
| 241 |
cfg['model']['image_size'] = 512
|
|
@@ -281,6 +285,7 @@ models = [
|
|
| 281 |
ddgan_sd_v10,
|
| 282 |
ddgan_sd_v11,
|
| 283 |
ddgan_laion2b_v2,
|
|
|
|
| 284 |
]
|
| 285 |
|
| 286 |
def get_model(model_name):
|
|
|
|
| 236 |
cfg = ddgan_sd_v9()
|
| 237 |
return cfg
|
| 238 |
|
| 239 |
+
def ddgan_ddb_v1():
|
| 240 |
+
cfg = ddgan_sd_v9()
|
| 241 |
+
return cfg
|
| 242 |
+
|
| 243 |
def ddgan_sd_v11():
|
| 244 |
cfg = ddgan_sd_v10()
|
| 245 |
cfg['model']['image_size'] = 512
|
|
|
|
| 285 |
ddgan_sd_v10,
|
| 286 |
ddgan_sd_v11,
|
| 287 |
ddgan_laion2b_v2,
|
| 288 |
+
ddgan_ddb_v1,
|
| 289 |
]
|
| 290 |
|
| 291 |
def get_model(model_name):
|