| { | |
| "_class_name": "EnvDecoder", | |
| "_diffusers_version": "0.28.0", | |
| "_name_or_path": "outputs/env_encoding_latent/checkpoint-48000", | |
| "act_fn": "none", | |
| "channels": [ | |
| 1280, | |
| 640, | |
| 320, | |
| 320 | |
| ], | |
| "feat_input": false, | |
| "latent_encoder": true, | |
| "num_attention_heads": [ | |
| 20, | |
| 20, | |
| 10, | |
| 5 | |
| ], | |
| "num_res_blocks": 2, | |
| "only_cross_attention": true, | |
| "out_channels": 8 | |
| } | |