This commit is contained in:
lvmin
2023-08-10 10:48:08 -07:00
parent 669314dff1
commit c4f798221d
2 changed files with 8 additions and 2 deletions
+2 -1
View File
@@ -75,7 +75,8 @@ class KSamplerWithRefiner:
sigmas = self.calculate_sigmas(new_steps).to(self.device)
self.sigmas = sigmas[-(steps + 1):]
def sample(self, noise, positive, negative, cfg, latent_image=None, start_step=None, last_step=None,
def sample(self, noise, positive, negative, refiner_positive, refiner_negative, cfg, latent_image=None,
start_step=None, last_step=None, refiner_switch_step=None,
force_full_denoise=False, denoise_mask=None, sigmas=None, callback=None, disable_pbar=False, seed=None):
if sigmas is None:
sigmas = self.sigmas