suvadityamuk's picture
End of training
ae2cc14 verified
raw
history blame contribute delete
684 Bytes
{
"_class_name": "UVit2DModel",
"_diffusers_version": "0.27.0.dev0",
"_name_or_path": "amused/amused-512",
"attention_dropout": 0.0,
"block_num_heads": 12,
"block_out_channels": 768,
"codebook_size": 8192,
"cond_embed_dim": 768,
"downsample": true,
"encoder_hidden_size": 768,
"hidden_dropout": 0.0,
"hidden_size": 1024,
"in_channels": 768,
"intermediate_size": 2816,
"layer_norm_eps": 1e-06,
"ln_elementwise_affine": true,
"micro_cond_embed_dim": 1280,
"micro_cond_encode_dim": 256,
"num_attention_heads": 16,
"num_hidden_layers": 22,
"num_res_blocks": 3,
"sample_size": 32,
"upsample": true,
"use_bias": false,
"vocab_size": 8256
}