sayakpaul
/

tiny-sd-pipeline-for-single-file-testing

StableDiffusionPipeline

Model card Files Files and versions

sayakpaul HF Staff commited on Feb 13, 2024

Commit

e5d53ec

·

verified ·

1 Parent(s): 3fa9dca

Update config.yaml

Files changed (1) hide show

config.yaml +1 -1

config.yaml CHANGED Viewed

@@ -35,7 +35,7 @@ model:
         model_channels: 4
         channel_mult: [ 1, 2 ]
         attention_resolutions: [2, 1]
-        num_heads: 8
         num_res_blocks: 1
         norm_num_groups: 2
         use_spatial_transformer: True

         model_channels: 4
         channel_mult: [ 1, 2 ]
         attention_resolutions: [2, 1]
+        num_heads: 2
         num_res_blocks: 1
         norm_num_groups: 2
         use_spatial_transformer: True