Commit d52a1e1a authored by drhead's avatar drhead Committed by GitHub

lint

parent 39a6d565
...@@ -4,16 +4,16 @@ import torch ...@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment