(
self, model: ModelPatcher, cfg, stg_scale, rescale_scale: float = None
)
| 188 | |
| 189 | class STGGuider(comfy.samplers.CFGGuider): |
| 190 | def __init__( |
| 191 | self, model: ModelPatcher, cfg, stg_scale, rescale_scale: float = None |
| 192 | ): |
| 193 | model = model.clone() |
| 194 | super().__init__(model) |
| 195 | |
| 196 | self.stg_flag = STGFlag( |
| 197 | do_skip=False, |
| 198 | skip_layers=model.model_options["transformer_options"]["skip_block_list"], |
| 199 | ) |
| 200 | |
| 201 | self.patch_model(model, self.stg_flag) |
| 202 | |
| 203 | self.cfg = cfg |
| 204 | self.stg_scale = stg_scale |
| 205 | self.rescale_scale = rescale_scale |
| 206 | |
| 207 | @classmethod |
| 208 | def patch_model(cls, model: ModelPatcher, stg_flag: STGFlag): |
no test coverage detected