| 24 | raise NotImplementedError(f"Tensor detected that exceeds dims supported by C++ code! ({key} @ {data.shape})") |
| 25 | |
| 26 | class ModelFlux(ModelTemplate): |
| 27 | arch = "flux" |
| 28 | keys_detect = [ |
| 29 | ("transformer_blocks.0.attn.norm_added_k.weight",), |
| 30 | ("double_blocks.0.img_attn.proj.weight",), |
| 31 | ] |
| 32 | keys_banned = ["transformer_blocks.0.attn.norm_added_k.weight",] |
| 33 | |
| 34 | class ModelSD3(ModelTemplate): |
| 35 | arch = "sd3" |
nothing calls this directly
no outgoing calls
no test coverage detected