Delete ComfyUI-HyperSDXL1StepUnetScheduler
Browse files
ComfyUI-HyperSDXL1StepUnetScheduler/__init__.py
DELETED
@@ -1,2 +0,0 @@
|
|
1 |
-
from .node import NODE_CLASS_MAPPINGS
|
2 |
-
__all__ = ['NODE_CLASS_MAPPINGS']
|
|
|
|
|
|
ComfyUI-HyperSDXL1StepUnetScheduler/node.py
DELETED
@@ -1,31 +0,0 @@
|
|
1 |
-
import comfy.samplers
|
2 |
-
import comfy.sample
|
3 |
-
from comfy.k_diffusion import sampling as k_diffusion_sampling
|
4 |
-
import latent_preview
|
5 |
-
import torch
|
6 |
-
import comfy.utils
|
7 |
-
|
8 |
-
|
9 |
-
class HyperSDXL1StepUnetScheduler:
|
10 |
-
@classmethod
|
11 |
-
def INPUT_TYPES(s):
|
12 |
-
return {"required":
|
13 |
-
{"model": ("MODEL",),
|
14 |
-
"steps": ("INT", {"default": 1, "min": 1, "max": 10}),
|
15 |
-
}
|
16 |
-
}
|
17 |
-
RETURN_TYPES = ("SIGMAS",)
|
18 |
-
CATEGORY = "sampling/custom_sampling/schedulers"
|
19 |
-
|
20 |
-
FUNCTION = "get_sigmas"
|
21 |
-
|
22 |
-
def get_sigmas(self, model, steps):
|
23 |
-
timesteps = torch.tensor([800])
|
24 |
-
sigmas = model.model.model_sampling.sigma(timesteps)
|
25 |
-
sigmas = torch.cat([sigmas, sigmas.new_zeros([1])])
|
26 |
-
return (sigmas, )
|
27 |
-
|
28 |
-
|
29 |
-
NODE_CLASS_MAPPINGS = {
|
30 |
-
"HyperSDXL1StepUnetScheduler": HyperSDXL1StepUnetScheduler,
|
31 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|