Upload StableAudioPipeline

scheduler/scheduler_config.json

@@ -14,12 +14,15 @@
   "prediction_type": "epsilon",
   "rescale_betas_zero_snr": false,
   "sample_max_value": 1.0,
-  "solver_order": 3,
+  "sigma_max": 500,
+  "sigma_min": 0.3,
+  "solver_order": 2,
   "solver_type": "midpoint",
   "steps_offset": 0,
   "thresholding": false,
   "timestep_spacing": "linspace",
   "trained_betas": null,
+  "use_exponential_sigmas": true,
   "use_karras_sigmas": false,
   "use_lu_lambdas": false,
   "variance_type": null

tokenizer/tokenizer_config.json

@@ -930,8 +930,7 @@
   "clean_up_tokenization_spaces": true,
   "eos_token": "</s>",
   "extra_ids": 100,
-  "max_length": 128,
-  "model_max_length": 1000000000000000019884624838656,
+  "model_max_length": 128,
   "pad_token": "<pad>",
   "tokenizer_class": "T5Tokenizer",
   "truncation": true,

transformer/config.json

@@ -10,5 +10,6 @@
   "num_key_value_attention_heads": 12,
   "num_layers": 24,
   "out_channels": 64,
+  "sample_size": 1024.0,
   "timestep_features_dim": 256
 }

vae/config.json

@@ -18,5 +18,6 @@
     8,
     8
   ],
-  "encoder_hidden_size": 128
+  "encoder_hidden_size": 128,
+  "sampling_rate": 44100
 }