mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2025-02-05 13:11:14 +01:00
Revert whitespace fix
This commit is contained in:
parent
5a9c8e1d87
commit
3b0ace3410
@ -17,7 +17,7 @@ index b967b55..35ef520 100644
|
|||||||
# sampling
|
# sampling
|
||||||
+ if sampler in ('ddim', 'dpm2', 'heun', 'dpm2_a', 'lms') and not hasattr(self, 'ddim_timesteps'):
|
+ if sampler in ('ddim', 'dpm2', 'heun', 'dpm2_a', 'lms') and not hasattr(self, 'ddim_timesteps'):
|
||||||
+ self.make_schedule(ddim_num_steps=S, ddim_eta=eta, verbose=False)
|
+ self.make_schedule(ddim_num_steps=S, ddim_eta=eta, verbose=False)
|
||||||
|
|
||||||
if sampler == "plms":
|
if sampler == "plms":
|
||||||
self.make_schedule(ddim_num_steps=S, ddim_eta=eta, verbose=False)
|
self.make_schedule(ddim_num_steps=S, ddim_eta=eta, verbose=False)
|
||||||
@@ -528,39 +530,46 @@ class UNet(DDPM):
|
@@ -528,39 +530,46 @@ class UNet(DDPM):
|
||||||
@ -103,7 +103,7 @@ index b967b55..35ef520 100644
|
|||||||
@@ -730,10 +740,13 @@ class UNet(DDPM):
|
@@ -730,10 +740,13 @@ class UNet(DDPM):
|
||||||
unconditional_guidance_scale=unconditional_guidance_scale,
|
unconditional_guidance_scale=unconditional_guidance_scale,
|
||||||
unconditional_conditioning=unconditional_conditioning)
|
unconditional_conditioning=unconditional_conditioning)
|
||||||
|
|
||||||
+ if callback: yield from callback(i)
|
+ if callback: yield from callback(i)
|
||||||
+ if img_callback: yield from img_callback(x_dec, i)
|
+ if img_callback: yield from img_callback(x_dec, i)
|
||||||
+
|
+
|
||||||
@ -239,7 +239,7 @@ index b967b55..35ef520 100644
|
|||||||
|
|
||||||
-
|
-
|
||||||
+ if img_callback: yield from img_callback(x, i)
|
+ if img_callback: yield from img_callback(x, i)
|
||||||
|
|
||||||
d = to_d(x, sigma_hat, denoised)
|
d = to_d(x, sigma_hat, denoised)
|
||||||
# Midpoint method, where the midpoint is chosen according to a rho=3 Karras schedule
|
# Midpoint method, where the midpoint is chosen according to a rho=3 Karras schedule
|
||||||
@@ -945,11 +982,13 @@ class UNet(DDPM):
|
@@ -945,11 +982,13 @@ class UNet(DDPM):
|
||||||
|
Loading…
Reference in New Issue
Block a user