| 91 | save_file(fsd, path) |
| 92 | |
| 93 | class ModelWan(ModelHyVid): |
| 94 | arch = "wan" |
| 95 | keys_detect = [ |
| 96 | ( |
| 97 | "blocks.0.self_attn.norm_q.weight", |
| 98 | "text_embedding.2.weight", |
| 99 | "head.modulation", |
| 100 | ) |
| 101 | ] |
| 102 | keys_hiprec = [ |
| 103 | ".modulation" # nn.parameter, can't load from BF16 ver |
| 104 | ] |
| 105 | |
| 106 | class ModelLTXV(ModelTemplate): |
| 107 | arch = "ltxv" |
nothing calls this directly
no outgoing calls
no test coverage detected