Pixel-1 / config.json
Raziel1234's picture
Update config.json
ed41163 verified
{
"architectures": [
"TopAIImageGenerator"
],
"model_type": "topai_image_gen",
"auto_map": {
"AutoConfig": "configuration_pixel.TopAIImageConfig",
"AutoModel": "modeling_pixel.TopAIImageGenerator"
},
"input_dim": 768,
"hidden_dim": 512,
"image_channels": 3,
"resolution": 128,
"torch_dtype": "float32",
"transformers_version": "5.2.0",
"initializer_range": 0.02,
"model_name": "Pixel-1"
}