This commit is contained in:
drhead
2024-06-08 18:56:23 -04:00
committed by GitHub
parent 39a6d5655f
commit d52a1e1a22

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