Skip to content

Commit 47dfc92

Browse files
author
pnsuau
committed
fix: add new args to define_g
1 parent 5623e31 commit 47dfc92

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Diff for: models/diffusion_networks.py

+2
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ def define_G(
3030
G_unet_mha_norm_layer,
3131
G_unet_mha_group_norm_size,
3232
alg_palette_sampling_method,
33+
alg_palette_scaling_factor,
34+
alg_palette_normalize_yt,
3335
dropout=0,
3436
channel_mults=(1, 2, 4, 8),
3537
conv_resample=True,

0 commit comments

Comments
 (0)