Mehdi Cherti commited on
Commit
7ac3881
1 Parent(s): b991bac

update list of models

Browse files
Files changed (1) hide show
  1. run.py +17 -0
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):