Spaces:
Runtime error
Runtime error
Mehdi Cherti
commited on
Commit
•
7ac3881
1
Parent(s):
b991bac
update list of models
Browse files
run.py
CHANGED
@@ -227,6 +227,20 @@ def ddgan_sd_v9():
|
|
227 |
cfg['model']['text_encoder'] = "openclip/ViT-H-14/laion2b_s32b_b79k"
|
228 |
return cfg
|
229 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
230 |
models = [
|
231 |
ddgan_cifar10_cond17, # cifar10, cross attn for discr
|
232 |
ddgan_cifar10_cond18, # cifar10, xl encoder
|
@@ -264,6 +278,9 @@ models = [
|
|
264 |
ddgan_sd_v7,
|
265 |
ddgan_sd_v8,
|
266 |
ddgan_sd_v9,
|
|
|
|
|
|
|
267 |
]
|
268 |
|
269 |
def get_model(model_name):
|
|
|
227 |
cfg['model']['text_encoder'] = "openclip/ViT-H-14/laion2b_s32b_b79k"
|
228 |
return cfg
|
229 |
|
230 |
+
def ddgan_sd_v10():
|
231 |
+
cfg = ddgan_sd_v9()
|
232 |
+
cfg['model']['num_timesteps'] = 2
|
233 |
+
return cfg
|
234 |
+
|
235 |
+
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
|
242 |
+
return cfg
|
243 |
+
|
244 |
models = [
|
245 |
ddgan_cifar10_cond17, # cifar10, cross attn for discr
|
246 |
ddgan_cifar10_cond18, # cifar10, xl encoder
|
|
|
278 |
ddgan_sd_v7,
|
279 |
ddgan_sd_v8,
|
280 |
ddgan_sd_v9,
|
281 |
+
ddgan_sd_v10,
|
282 |
+
ddgan_sd_v11,
|
283 |
+
ddgan_laion2b_v2,
|
284 |
]
|
285 |
|
286 |
def get_model(model_name):
|