Upload StableDiffusionPipeline

scheduler/scheduler_config.json

@@ -1,9 +1,9 @@
 {
   "_class_name": "DDPMScheduler",
   "_diffusers_version": "0.30.1",
-  "beta_end": 0.02,
-  "beta_schedule": "linear",
-  "beta_start": 0.0001,
+  "beta_end": 0.012,
+  "beta_schedule": "scaled_linear",
+  "beta_start": 0.00085,
   "clip_sample": false,
   "clip_sample_range": 1.0,
   "dynamic_thresholding_ratio": 0.995,
@@ -11,6 +11,8 @@
   "prediction_type": "epsilon",
   "rescale_betas_zero_snr": false,
   "sample_max_value": 1.0,
+  "set_alpha_to_one": false,
+  "skip_prk_steps": true,
   "steps_offset": 1,
   "thresholding": false,
   "timestep_spacing": "leading",

unet/config.json

@@ -5,7 +5,7 @@
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
   "addition_time_embed_dim": null,
-  "attention_head_dim": 32,
+  "attention_head_dim": 16,
   "attention_type": "default",
   "block_out_channels": [
     32
@@ -28,7 +28,7 @@
   "flip_sin_to_cos": true,
   "freq_shift": 0,
   "in_channels": 4,
-  "layers_per_block": 16,
+  "layers_per_block": 8,
   "mid_block_only_cross_attention": null,
   "mid_block_scale_factor": 1,
   "mid_block_type": "UNetMidBlock2DCrossAttn",
@@ -43,7 +43,7 @@
   "resnet_skip_time_act": false,
   "resnet_time_scale_shift": "default",
   "reverse_transformer_layers_per_block": null,
-  "sample_size": 2048,
+  "sample_size": 1024,
   "time_cond_proj_dim": null,
   "time_embedding_act_fn": null,
   "time_embedding_dim": null,

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2ea05af2a8f423e8c0ffc7c4f493ab00bb793e181a39ee1e7c00ea7924c8284d
-size 4374960
+oid sha256:f4811c6ac1d572de5ac7281804906bcf1650e252337193f88873dc12f9e89bc9
+size 2537056