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

update models

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