minor

model_configs.py

@@ -125,8 +125,8 @@ def ddgan_cc12m_v14():
 @register
 def ddgan_cc12m_v15():
     cfg = ddgan_cc12m_v11()
-    cfg['model']['mismatch_loss'] = ''
-    cfg['model']['grad_penalty_cond'] = ''
+    cfg['model']['mismatch_loss'] = True
+    cfg['model']['grad_penalty_cond'] = True
     return cfg
 
 @register
@@ -150,8 +150,8 @@ def ddgan_cifar10_cond18():
 def ddgan_cifar10_cond19():
     cfg = ddgan_cifar10_cond17()
     cfg['model']['discr_type'] = 'small_cond_attn'
-    cfg['model']['mismatch_loss'] = ''
-    cfg['model']['grad_penalty_cond'] = ''
+    cfg['model']['mismatch_loss'] = True
+    cfg['model']['grad_penalty_cond'] =True
     return cfg
 
 @register
@@ -170,8 +170,8 @@ def ddgan_laion_aesthetic_v2():
 def ddgan_laion_aesthetic_v3():
     cfg = ddgan_laion_aesthetic_v1()
     cfg['model']['text_encoder'] = "google/t5-v1_1-xl" 
-    cfg['model']['mismatch_loss'] = ''
-    cfg['model']['grad_penalty_cond'] = ''
+    cfg['model']['mismatch_loss'] = True
+    cfg['model']['grad_penalty_cond'] = True
     return cfg
 
 @register
@@ -184,8 +184,8 @@ def ddgan_laion_aesthetic_v4():
 @register
 def ddgan_laion_aesthetic_v5():
     cfg = ddgan_laion_aesthetic_v1()
-    cfg['model']['mismatch_loss'] = ''
-    cfg['model']['grad_penalty_cond'] = ''
+    cfg['model']['mismatch_loss'] = True
+    cfg['model']['grad_penalty_cond'] = True
     return cfg
 
 
@@ -193,8 +193,8 @@ def ddgan_laion_aesthetic_v5():
 @register
 def ddgan_laion2b_v1():
     cfg = ddgan_laion_aesthetic_v3()
-    cfg['model']['mismatch_loss'] = ''
-    cfg['model']['grad_penalty_cond'] = ''
+    cfg['model']['mismatch_loss'] = True
+    cfg['model']['grad_penalty_cond'] = True
     cfg['model']['num_channels_dae'] = 224
     cfg['model']['batch_size'] = 2
     cfg['model']['discr_type'] = "large_cond_attn"
@@ -302,6 +302,7 @@ def ddgan_laion_aesthetic_v14():
 def ddgan_sd_v9():
     cfg = ddgan_laion_aesthetic_v3()
     cfg['model']['text_encoder'] = "openclip/ViT-H-14/laion2b_s32b_b79k" 
+    cfg['model']['classifier_free_guidance_proba'] = 0.0
     return cfg
 
 @register