xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-06 18:07:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add type hints

Browse Source
This commit is contained in:
catboxanon 2025-05-01 18:38:43 -04:00
parent c544038d6f
commit 65565c1d21
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_custom_sampler.py
View File

@ -266,7 +266,7 @@ class ExtendIntermediateSigmas:
FUNCTION = "extend"
def extend(self, sigmas, steps, start_at_sigma, end_at_sigma, spacing):
def extend(self, sigmas: torch.Tensor, steps: int, start_at_sigma: float, end_at_sigma: float, spacing: str):
if start_at_sigma < 0:
start_at_sigma = float("inf")