| 62 | ] |
| 63 | |
| 64 | class CosmosPredict2(ModelTemplate): |
| 65 | arch = "cosmos" |
| 66 | keys_detect = [ |
| 67 | ( |
| 68 | "blocks.0.mlp.layer1.weight", |
| 69 | "blocks.0.adaln_modulation_cross_attn.1.weight", |
| 70 | ) |
| 71 | ] |
| 72 | keys_hiprec = ["pos_embedder"] |
| 73 | keys_ignore = ["_extra_state", "accum_"] |
| 74 | |
| 75 | class ModelHyVid(ModelTemplate): |
| 76 | arch = "hyvid" |
nothing calls this directly
no outgoing calls
no test coverage detected