This commit is contained in:
drhead 2024-06-08 18:56:23 -04:00 committed by GitHub
parent 39a6d5655f
commit d52a1e1a22
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions

View File

@ -4,16 +4,16 @@ import torch
import k_diffusion import k_diffusion
import numpy as np
from modules import shared
def to_d(x, sigma, denoised): def to_d(x, sigma, denoised):
"""Converts a denoiser output to a Karras ODE derivative.""" """Converts a denoiser output to a Karras ODE derivative."""
return (x - denoised) / sigma return (x - denoised) / sigma
k_diffusion.sampling.to_d = to_d k_diffusion.sampling.to_d = to_d
import numpy as np
from modules import shared
@dataclasses.dataclass @dataclasses.dataclass
class Scheduler: class Scheduler:
name: str name: str