(
self,
model: ModelPatcher,
cfg_scale,
eta,
norm_threshold,
momentum_coefficient: float = -0.9,
)
| 796 | |
| 797 | class APGGuider(comfy.samplers.CFGGuider): |
| 798 | def __init__( |
| 799 | self, |
| 800 | model: ModelPatcher, |
| 801 | cfg_scale, |
| 802 | eta, |
| 803 | norm_threshold, |
| 804 | momentum_coefficient: float = -0.9, |
| 805 | ): |
| 806 | self.model = model.clone() |
| 807 | super().__init__(self.model) |
| 808 | self.momentum_coefficient = momentum_coefficient |
| 809 | self.momentum_buffer = MomentumBuffer(self.momentum_coefficient) |
| 810 | self.eta = eta |
| 811 | self.norm_threshold = norm_threshold |
| 812 | self.cfg_scale = cfg_scale |
| 813 | self.previous_timestep = None |
| 814 | |
| 815 | def set_conds(self, positive, negative): |
| 816 | self.inner_set_conds({"positive": positive, "negative": negative}) |
nothing calls this directly
no test coverage detected