File size: 1,109 Bytes
8d31176 a5318e5 8d31176 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | {
"_name_or_path": "solintellegence/SolPix",
"license": "apache-2.0",
"release": "v1.0.0",
"architectures": [
"SolPixTransformer2D"
],
"model_type": "solpix",
"model_config": {
"latent_channels": 32,
"text_dim": 768,
"max_text_tokens": 96,
"width": 512,
"depth": 15,
"heads": 8,
"mlp_hidden": 1152,
"encoder_depth": 7,
"latent_downsample_factor": 32,
"image_channels": 3,
"repa_layer": 7
},
"transformer_class": "solpix.SolPixTransformer2D",
"prediction_type": "flow_velocity",
"text_encoder": {
"class": "transformers.T5EncoderModel",
"model_id": "google/flan-t5-base",
"hidden_size": 768,
"max_tokens": 96
},
"autoencoder": {
"class": "solpix.AutoencoderDCSol",
"implementation": "diffusers.AutoencoderDC",
"model_id": "mit-han-lab/dc-ae-f32c32-sana-1.1-diffusers",
"revision": "df0d9d634aea77793c1fb685d4b9db092c99e686",
"latent_channels": 32,
"downsample_factor": 32
},
"checkpoint": {
"filename": "step_00210000.pt",
"optimizer_step": 210000,
"ema_weights_key": "ema"
}
}
|