renyuxi commited on
Commit
3f00860
1 Parent(s): b6f665c

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