lint
This commit is contained in:
parent
39a6d5655f
commit
d52a1e1a22
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue