fix: add new args to define_g

jolibrain · Mar 16, 2023 · 47dfc92 · 47dfc92
1 parent 5623e31
commit 47dfc92
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/diffusion_networks.py b/models/diffusion_networks.py
@@ -30,6 +30,8 @@ def define_G(
     G_unet_mha_norm_layer,
     G_unet_mha_group_norm_size,
     alg_palette_sampling_method,
+    alg_palette_scaling_factor,
+    alg_palette_normalize_yt,
     dropout=0,
     channel_mults=(1, 2, 4, 8),
     conv_resample=True,