| { | |
| "cond_drop_prob": 0.1, | |
| "device": "cuda", | |
| "frozen_clip": true, | |
| "heads": 8, | |
| "init_scale": 0.25, | |
| "init_with_modified_layer": true, | |
| "input_channels": 6, | |
| "layers": 12, | |
| "n_ctx": 2048, | |
| "output_channels": 12, | |
| "pointe_cache_dir": "cache/point_e_model", | |
| "time_token_cond": true, | |
| "token_cond": true, | |
| "width": 512 | |
| } |