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 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|