Initial commit

config.json

@@ -0,0 +1 @@
+{"sample_size": 64, "in_channels": 8, "out_channels": 4, "down_block_types": ["CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "DownBlock2D"], "up_block_types": ["UpBlock2D", "CrossAttnUpBlock2D", "CrossAttnUpBlock2D", "CrossAttnUpBlock2D"], "mid_block_type": "UNetMidBlock2DCrossAttn", "only_cross_attention": false, "block_out_channels": [320, 640, 1280, 1280], "layers_per_block": 2, "attention_head_dim": 8, "num_attention_heads": null, "cross_attention_dim": 768, "dropout": 0.0, "use_linear_projection": false, "flip_sin_to_cos": true, "freq_shift": 0, "use_memory_efficient_attention": false, "split_head_dim": false, "transformer_layers_per_block": 1, "addition_embed_type": null, "addition_time_embed_dim": null, "addition_embed_type_num_heads": 64, "projection_class_embeddings_input_dim": null, "_use_default_values": ["use_memory_efficient_attention", "split_head_dim"], "_class_name": "FlaxUNet2DConditionModel", "_diffusers_version": "0.28.0.dev0", "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "attention_type": "default", "center_input_sample": false, "class_embed_type": null, "class_embeddings_concat": false, "conv_in_kernel": 3, "conv_out_kernel": 3, "cross_attention_norm": null, "downsample_padding": 1, "dual_cross_attention": false, "encoder_hid_dim": null, "encoder_hid_dim_type": null, "mid_block_only_cross_attention": null, "mid_block_scale_factor": 1, "norm_eps": 1e-05, "norm_num_groups": 32, "num_class_embeds": null, "resnet_out_scale_factor": 1.0, "resnet_skip_time_act": false, "resnet_time_scale_shift": "default", "reverse_transformer_layers_per_block": null, "time_cond_proj_dim": null, "time_embedding_act_fn": null, "time_embedding_dim": null, "time_embedding_type": "positional", "timestep_post_act": null, "upcast_attention": false}

diffusion_flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a47d5ef63312a001e29a59ef2b73a3366e78ae1fd2eb4bd27048a76c430f2c2
+size 3438154451