Rename sequential as private variable

This commit is contained in:
Sergey Borisov 2024-07-17 03:24:11 +03:00
parent b7c6c63005
commit cd1bc1595a

View File

@ -19,7 +19,7 @@ class StableDiffusionBackend:
self.unet = unet self.unet = unet
self.scheduler = scheduler self.scheduler = scheduler
config = get_config() config = get_config()
self.sequential_guidance = config.sequential_guidance self._sequential_guidance = config.sequential_guidance
def latents_from_embeddings(self, ctx: DenoiseContext, ext_manager: ExtensionsManager): def latents_from_embeddings(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
if ctx.inputs.init_timestep.shape[0] == 0: if ctx.inputs.init_timestep.shape[0] == 0:
@ -67,7 +67,7 @@ class StableDiffusionBackend:
# Note: The current handling of conditioning doesn't feel very future-proof. # Note: The current handling of conditioning doesn't feel very future-proof.
# This might change in the future as new requirements come up, but for now, # This might change in the future as new requirements come up, but for now,
# this is the rough plan. # this is the rough plan.
if self.sequential_guidance: if self._sequential_guidance:
ctx.negative_noise_pred = self.run_unet(ctx, ext_manager, "negative") ctx.negative_noise_pred = self.run_unet(ctx, ext_manager, "negative")
ctx.positive_noise_pred = self.run_unet(ctx, ext_manager, "positive") ctx.positive_noise_pred = self.run_unet(ctx, ext_manager, "positive")
else: else: