| { | |
| "_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 | |
| } | |