(model: ModelPatcher)
| 215 | |
| 216 | @staticmethod |
| 217 | def get_transformer_blocks(model: ModelPatcher): |
| 218 | diffusion_model = model.get_model_object("diffusion_model") |
| 219 | key = "diffusion_model.transformer_blocks" |
| 220 | if diffusion_model.__class__.__name__ == "LTXVTransformer3D": |
| 221 | key = "diffusion_model.transformer.transformer_blocks" |
| 222 | return model.get_model_object(key) |
| 223 | |
| 224 | def set_conds(self, positive, negative): |
| 225 | self.inner_set_conds({"positive": positive, "negative": negative}) |