(x_t, sigma)
| 168 | if guidance_scale != 0 and guidance_scale != 1: |
| 169 | |
| 170 | def guided_denoiser(x_t, sigma): |
| 171 | x_t = th.cat([x_t, x_t], dim=0) |
| 172 | sigma = th.cat([sigma, sigma], dim=0) |
| 173 | x_0 = denoiser(x_t, sigma) |
| 174 | cond_x_0, uncond_x_0 = th.split(x_0, len(x_0) // 2, dim=0) |
| 175 | x_0 = uncond_x_0 + guidance_scale * (cond_x_0 - uncond_x_0) |
| 176 | return x_0 |
| 177 | |
| 178 | else: |
| 179 | guided_denoiser = denoiser |
nothing calls this directly
no test coverage detected