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

View File

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