config.json

{
  "_name_or_path": "openai/jukebox-1b-lyrics",
  "act_fn": "relu",
  "architectures": [
    "JukeboxVQVAE"
  ],
  "commit": 0.02,
  "conv_input_shape": 1,
  "conv_res_scale": false,
  "embed_dim": 64,
  "hop_fraction": [
    0.125,
    0.5,
    0.5
  ],
  "init_scale": 0.2,
  "levels": 3,
  "lmu": 0.99,
  "model_type": "jukebox_vqvae",
  "multipliers": [
    2,
    1,
    1
  ],
  "nb_discrete_codes": 2048,
  "res_conv_depth": 4,
  "res_conv_width": 32,
  "res_convolution_multiplier": 1,
  "res_dilation_cycle": null,
  "res_dilation_growth_rate": 3,
  "res_downs_t": [
    3,
    2,
    2
  ],
  "res_strides_t": [
    2,
    2,
    2
  ],
  "sample_length": 1058304,
  "torch_dtype": "float16",
  "transformers_version": "4.25.0.dev0",
  "zero_out": false
}