SupraLabs
/

SupraMNST-IMG-200k

digit_diffusion

Model card Files Files and versions

Harley-ml commited on 4 days ago

Commit

0b6a7dc

·

verified ·

1 Parent(s): 693e568

Update configuration.py

Files changed (1) hide show

configuration.py +2 -2

configuration.py CHANGED Viewed

@@ -18,8 +18,8 @@ class DigitDiffusionConfig(PretrainedConfig):
         in_channels: int = 1,
         out_channels: int = 1,
         num_classes: int = 10,
-        block_out_channels: Iterable[int] = (12, 12),
-        layers_per_block: int = 5,
         norm_num_groups: int = 4,
         cross_attention_dim: int = 8,
         class_embed_type: str | None = None,

         in_channels: int = 1,
         out_channels: int = 1,
         num_classes: int = 10,
+        block_out_channels: Iterable[int] = (12, 16),
+        layers_per_block: int = 8,
         norm_num_groups: int = 4,
         cross_attention_dim: int = 8,
         class_embed_type: str | None = None,