From 689a29809f2b5f5e7ff88d6ed9c738cad27ca893 Mon Sep 17 00:00:00 2001
From: Fractal <Fractal@FRACTAL>
Date: Sun, 30 Nov 2025 02:24:49 +0100
Subject: [PATCH] Add start and end step

---
 .vs/ComfyUI-EulerDiscreteScheduler/v17/.wsuo  | Bin 0 -> 14848 bytes
 .../v17/DocumentLayout.json                   |  23 ++++++++++
 .vs/ProjectSettings.json                      |   3 ++
 .vs/VSWorkspaceState.json                     |   7 +++
 .vs/slnx.sqlite                               | Bin 0 -> 90112 bytes
 __init__.py                                   |  40 +++++++++++++++---
 6 files changed, 68 insertions(+), 5 deletions(-)
 create mode 100644 .vs/ComfyUI-EulerDiscreteScheduler/v17/.wsuo
 create mode 100644 .vs/ComfyUI-EulerDiscreteScheduler/v17/DocumentLayout.json
 create mode 100644 .vs/ProjectSettings.json
 create mode 100644 .vs/VSWorkspaceState.json
 create mode 100644 .vs/slnx.sqlite

diff --git a/.vs/ComfyUI-EulerDiscreteScheduler/v17/.wsuo b/.vs/ComfyUI-EulerDiscreteScheduler/v17/.wsuo
new file mode 100644
index 0000000000000000000000000000000000000000..4355676d6b18d3b5d1e01d9a018defbb90a7d9d1
GIT binary patch
literal 14848
zcmeHNOLH4V5FR-R5Z)o+KnNg=0RxedWXrY;7{{;JRhXE_52_GFtzL;@$r6$r;v!UW
z<OUZ`T;P;57mgG~ao}+X+&PdR0XJg4Z&o9X*3xRVGT2LLRIQnvot>WU>F(*CnO|?7
z_~WauPu~)jG$oFR+q=ib9a_ClD~kt(IEiQ0-QL~Z<tvs^>?IFz;G!ttR}p?OFSd}E
z#15y@aq1VxYIA;no3B1Q_171l8^8PRM!Sz6G8?Y|V?kUIOTeB%X<l4w(@W#)e$iLM
zPqsI{ZT_@`YKX5)3P14{MGhk<i8LSvUKoIMwOy-hQ1toq>2B3dub;xhX0_si*anuI
z$bx>tB@Oc&YD)TtP0GZ76!pgdPU)XVTcy_zqx`o{K$G$#XmbzXUch|-&hrBR^6&(J
zQ~5Z6cA&qSKZ=}Pov!A`tMyH~?^PWj2mVX_FaGprP5-A4Q2!Spi#`1Z6OfeX-H9^o
z0CkwQ;S_*2h0k{(Z>Rr{p^fTG<<xnW*+(xOP7ZKxngQj3^_)Yl*X`z?z982Ut`%GZ
zC_h{SC>LA<xIUZ#JOOwT@D#ugcp5MOcm^;Ccoy&+;Ca9h;03^ofR_Mg0Rg};;AKD%
z5CVh&BY;sr1TY5BX`00IE0(&~wdWbFtbeD<hjZrkXH>KBJq%ICZynx7T6`d8;RDeV
zOw0c=cw?8~e>CX5U0d~zd{yN+;7Py-x`ZG7t_|sRCB<btwZqLm{4VgnhVLk=Oji6e
z(*HBypB2FedSn%G9(4tbvDjop;i~^fx#x_?PTTywiucqhCMWpq_%H<g$`^BrKMK5`
zNbGy@(+^I9_eId1m6BkGox0k&|2qNPTvM2=^b@}+dnt?}CF23R@r(gK*Q=8Nr}U4b
zZ7ci>a?Vrm!a3t5$qO`ZnMnT;yrp046hCo)NrQ@HrN54UN93Uy1?Z<;0{*V_Zw43&
zkoG+Eu7uoa+@r>H2K4*z!1cl@{c~tLRPA6@TAxkv2KZja%*UYljKM14JR>7}>EoKw
z=j%gXj7^;4pGDjH{1+sqJfw}fzEPe30xaPQMpDMJohAf<pHoSFc8dQT+Fl1RUBDb)
z5zDZCr9GB+T~Jf((*W>ujV3)#@lT^IdBzk2^+`}l3wIT?QtL`sVJZJPIRmB@7{&}0
zFjpx#%5BLjem~_x{uc0^;(sUjxlUCCEwlmrTqA^-c7gvqzN5|9uSCA9v|oL8(vQN;
zyp!+PR+sl>$ycMg5)Z>8pe$4bbu>}aP=1j@?PaN(;J;7z{hi>ay;SS39sUjJ1#Ce_
zic&kd9#|9BUn%#B2iEt^o+<k({<FhB3;CzS6<{6fKERvc2Cao@nJEACbxDs?{?lI4
zUNJevzlu>%BWa6tKV%a8XDF@qTV$sW{uF-Z9FN=7?f0CAZt#=u{J!2taT<?~Q@7)H
zjL4>;vc-CnZtC=YoR;ZBegl{llOkSS54a{#*8PHBNlwaq9BWk;wwLQg8IXZQ`ep1;
zhPD`J<(EFhHguIMM;y^v5D-R*kwqRqt{z-fcTl&E+BhtXDcxn1m+`$>8$v`k4B-C=
zB0Kgpju!OFlX7+9Y8A(*zxU|K^WY&VDM>26^(Yl(%p_;dI}$|stzg7u<VjEQ-a7JY
z*zFkgGzE-1@{)odJR<+rIY`2K(IS~LCB%0gr6UU~I{|s3CjaPB8qAZlP0R>)FiflD
z6}ze=NLeNNo;gp?##LwqwW9<&vWOulSwDMJvPerw`Y!ar83B+Quh=Y}W#=&J45+BD
zd+OQibePn6cRu!dv=A*v$)`m=jy|w7I%;|D^pSDbb$}T+tij@P&wzG!0l5Kb<2?#m
zF?~lhE-s=Sci7p|j0-aGBG+&aB!E5NZG6FgcmskSGVf$?SD3p+JQ+yUT5!(0N!$l|
zN4)F03)0_B;=ZA|7okU@Yp7+<i$R1u`Y%v>vM=f1u>7wj&OKxB!Bip>G@@gvKy)HG
z9vIC;!vQ0cj0U2iw2?`KBH={DkVMJtVvZ*}&F7gwY;`h---n^(*k~e}3<uI9k#Jx%
zJdq9<5p)<dlIc)1n21J=v5&|l?jNc2_Dyw?qG~$J2eF^nxC34mjAzNT5aLe$+of*j
zU-kXo*l^_iUzXIil&Wr<=Z$hDUApy~+wavGr#0`L|2CeBtNq3QTW*#9n6%gJ@4YF(
z?4AGg&VO{BP;z0HAExu4Wq2ZMu=v*VE`86_jutoPuZ!rA>)3v!PWi9n?cM)qyno`s
z`yaKtxRe8ZZd;Y@?td)9yUxpqeW&4Us?|CCSr_MT^to57gR|a$FweuxQ#X43=MW_n
zaVEeWHl7)9FF{Yd|Iw-Z^IoHhdt$)B9Z1F%#6rxBO?c;o_W|BQq`^ppccaR+Gtl}7
zCS1_3-+!_@|62jpE&Leon9)@I+)<5y^z#lj@7y_^f6}h(kCp$u`#<Elr|$nq`B(Ss
zm0Z~6y8IPVbDs{}a34R|;px3Q{2Z|AzYF`{vHk7Y|N1(8(Dt9}0MGt8A~*XFJ^OL^
Jb=+=*{{WjYlMDa=

literal 0
HcmV?d00001

diff --git a/.vs/ComfyUI-EulerDiscreteScheduler/v17/DocumentLayout.json b/.vs/ComfyUI-EulerDiscreteScheduler/v17/DocumentLayout.json
new file mode 100644
index 0000000..8107429
--- /dev/null
+++ b/.vs/ComfyUI-EulerDiscreteScheduler/v17/DocumentLayout.json
@@ -0,0 +1,23 @@
+{
+  "Version": 1,
+  "WorkspaceRootPath": "C:\\Users\\Fractal\\source\\repos\\ComfyUI-EulerDiscreteScheduler\\",
+  "Documents": [],
+  "DocumentGroupContainers": [
+    {
+      "Orientation": 0,
+      "VerticalTabListWidth": 256,
+      "DocumentGroups": [
+        {
+          "DockedWidth": 200,
+          "SelectedChildIndex": -1,
+          "Children": [
+            {
+              "$type": "Bookmark",
+              "Name": "ST:0:0:{1c64b9c2-e352-428e-a56d-0ace190b99a6}"
+            }
+          ]
+        }
+      ]
+    }
+  ]
+}
\ No newline at end of file
diff --git a/.vs/ProjectSettings.json b/.vs/ProjectSettings.json
new file mode 100644
index 0000000..f8b4888
--- /dev/null
+++ b/.vs/ProjectSettings.json
@@ -0,0 +1,3 @@
+{
+  "CurrentProjectSetting": null
+}
\ No newline at end of file
diff --git a/.vs/VSWorkspaceState.json b/.vs/VSWorkspaceState.json
new file mode 100644
index 0000000..9ca4b2c
--- /dev/null
+++ b/.vs/VSWorkspaceState.json
@@ -0,0 +1,7 @@
+{
+  "ExpandedNodes": [
+    ""
+  ],
+  "SelectedNode": "\\C:\\Users\\Fractal\\Source\\Repos\\ComfyUI-EulerDiscreteScheduler",
+  "PreviewInSolutionExplorer": false
+}
\ No newline at end of file
diff --git a/.vs/slnx.sqlite b/.vs/slnx.sqlite
new file mode 100644
index 0000000000000000000000000000000000000000..1793f679d49febfe2034bbe3cbe58c8308c54ef7
GIT binary patch
literal 90112
zcmeI4U2GfKb%19$6e&?6N3vH-?<$FB>)l$cJ+k;u##o><HL|&;DVe1FQ<WfwGnX=V
zDN-g!UJFQ3$ZodCF1kP-`qa0k4=IqKXo12HMbf?$XbTigi@vl$7ARVvNzkG|3#18B
zpuKl~IKMQta@U4ehk!<$x#!+{&Ueo_bLU=?wVUORg1B~nuq!HDlA5Pzn)+pqqbTYO
z{(BYw8L!j$jo&!I|7pwnac|F1EA1ao5n=Y{MCKIx^SN)%&CgyBeLVBg^!I~I;3o5R
z-_Lwi-#g=)fa@;}foY=Y;hT|Q^fpf)-t6>b^ssiY+v<1MP`lIXD4l+<slDy(tarMo
zBsWW5LTs@r6zT$3s;mijxa}jg(Flh){Mx9v<s01cw!Pa%>~Jw0j25w)SH0D&h`Y#J
zxyS_N%$gHaB7zOBxKS<>TPqvILQRN$_3gP}l;`OWc|~k>k<)bNd01DbUbs>gJS-1~
zmNhOeh$Xp!`PPN2LY3RBmTnZPce(4rU9PZI-zZgZvp0lF9czXEw#wy;dMlfQ{)3K;
z2G-h!j3z<`p(?BkRiRQ8YTUL_L+B|T<)E=Vx`3OAnN{$(ZEda*5Ew(TuqN1w)`YTP
z78%ND04ui&)#9~6bvcnv$3_Y}xEfW~_q*MLoBLw7)9xVIs{#9X81=4WFEM!;78|7n
zNkYS6r`72a6GvE`Vh$g4$(bfw0g^ze9Sjw;J3{EZL{M^O?c=a4_a%|UXSM(ScmoWf
zV!yk;+jGl>L$Sk4v%zQ+$5m5}D^*iT?Df!~C`$KG)2+G|Wlq64dz3kg$F^gMIcFX0
zd+y#FvBz5>HR3*Ks1fI^<vgFYN1U@5_isd;v&b4VM~FCmK)rN!d5$gO)P5&B%(MB6
z5!P0t41c$2V+f9%*SO${fA1i#AMD{QX=Ey+3nTLCon16k#NE9HN0LXqbOUEnoUe!{
z@-xBcr5OE)RXx$@mbBL-2jhaKSV$k%txD<U*67Y-$c!pk+{-A_4a4^s<BF!TBaw_s
zZ=t~u2bwGMKcc3C(O8Urd|CBFqcN>{``L`$2knIx4>Ss_gLDMX(>!COux5^#W>+U>
zS)<>PsYj_yhV0A1U{v6ZA#*x%b5k6kp3)>~+gUd4IetzlYYUd4bc&G)gV){pBEFre
zDmW_;Up~AN2u9z-neV{Ne3mHGyLQ*K<UfmeP`5{)0y(QqXl;URPcdR~$~rA_bxWvt
zD;$`huvv40a$ewLkFWW)BwaL<w1t*a&nbI=IEyVI;w(B|sAxKyUVem`$k;J3bDf&U
z$b5>tjxi;8whUhy5;hI)O36J9EQ>>Bqcuc>2MAA#%*?CL7P&QHy|7iTbJG4mjabD<
z(>TccOfVW>q7ToiVNyei(&_D}F|E(HwJUzRKe^>pH&UnjT7_{z8{6~+q8BdEcU8|B
z-p6w^)EuJw`>2O!ZT3^f<~NSQ<sNqai@0X&@G2dQE-lfIklLoMoBq#dYhzo#YH>OJ
z%5(FyTVi(~kE6poxA3#tLMY6Bk7B|<px7UVe-Qr3OY@`&iUI*300e*l5C8%|00;m9
zAOHk_fR#Yh=cg_$l*kSnUXdeP6vh62Pl+cU-)5$$%fHEx$9T0^+g7S+3-!Y!Legp`
zA@T84CdsF=xm8|FN-~c!={Q==r?VN9<df~hYAT*h@u^H63!!w1&m}W)zSYX-rG%JI
zCvde?E18foxh$W^#U(x^Wn^AVh;cqC=d!6*TTI9)na{S8EjcYGN&N($N@Qg|mlsof
zHkD{6P$H4eNC`gQYUlE;yu_otj9XgGCUHy244*{VRwA2gr{hwVmokZ5CM!vNCM)Ln
zR9?#PVqRY56D`zEX7X~b)y6`tb}E-nW%<<>%3z)31nw9cE5#FXJDHXd%IA4GpNJ=<
zw9MzTX%V-CCB<wy$tT+BT&C4-wO6xQUT#a-bUvQvbFzpPZ>1!@C8yGSJd>6aEvcPp
zV;(3SPiIo7&8N~i0<#_GbJ;{2BgnSXX%Xc!m<*D0@l-0G<&&%FG$ty=`FtC#^0+fe
z%BPd7$@nVY&ZebYwzbO3GKQUMXIA-q1`mrQx8>DL4jYw7;Nd@WiFFNm{#DvfosZV=
z4$Q8ougb<(k}K&juHa+w^Z$S01H6C$5C8%|00;m9AOHk_01yBIKmZ5;ftQcKD?Wy&
zogZ*nn5CH|TK^0J6T((8>~|>kAK34(|MBvf2pR(cAOHk_01yBIKmZ5;0U!VbfB+Eq
z+!I)0F4JQ^!oe&umuT0=0DSYz`5D`X0LcFTLinc?`)};uv46_`9{cO;9riNIvA)QU
zB7YhAcI3ArgGe#*MJxg@AOHk_01yBIKmZ5;0U!VbfB+D92??BA2vBKyCV;;e@Gg=R
zeLHx#^6s$TW2Qg3^sVncx%cGF`2dy0LbqzS`-As}d!mGDil`vF6wd~z5<ML#_IKL{
zTP0rD@1nt4XDAJjg2=<BOfHNXeJv88&QUXgCjNn-j?!$d>>U{WeJLEEcwF==JpMVV
zs{&u&s$SVx*_Dm**XIJ%Te$pvtWCSye}84Kw_}ujX;!N{z+dd|4A3tAosi**@=!6#
zqoDwG7T3i;EVgz-ur&IGnE(~T1$zg38jBUBzuPscy*90BTP_uaN=;a)->Dl#OThqj
z9#_2A*}2!nf6B_cdn0w;3IwRjxUk(7hxeMIqF{Z<ub=h$GO}2FZVD47Vx6JV+(AA3
zy9DabA>e^9+j`Bf>3|;M-?{1H>Za8TOn^E~O$SyU3=Or;_yW`$xXG0r{CjIVz5W0h
zC8y{BwMc|<iE)kW|1U(IQ0!0HAF=<$h9f^_|CW8i{to*!_8{_vZR6E|V&r>lg8kx4
z=;WX>5C8%|00;m9AOHk_01yBIKmZ85>;%p*Y1(V)V&OF=OM5O?%>M#YqMyD*!LrOb
z+OjMWc?Exb-nkYLo@d^otqTxyM9R79FdJsh(xat@&?zQH+t(LnNMF`Pg$3q3ZCgnQ
zlHL6=%Lf5srstZ$6lvG8P~Z<Tr)j)8z(klgXmd%x7i1P^$f^4L&;CBe{tx@#?0>TV
z&i;`7XZ92PZGgXL|Azf5_RrZrWB-KxWA+cQ2)uv*5C8%|00;m9AOHk_01yBIKmZ5;
zfzO)26hqU@f_j`+kF1}e=m<H4$zhHhX2~H$4m0F1O%6eF2#~`RIrzzeAqOABP&3;8
z|2HW7^#8ByCz1b+{3H9<>>she%|2r9N4~)}*b1AAe3$)2cK)+A5ZVU<KmZ5;0U!Vb
zfB+Bx0zd!=0D%)u;B|kR_T21e7MA>3+Izd7nSaw?q9<(lGwi(o9A(?`U-a{od(VHt
z{}yH6^jGCvyZ#aXS<15Qf7&0TocsQ2kJgR<Q~vXmW9R=1{>#+Zt^e2kSzPG#eSs;p
zZR_U$wEr}v?*6m>H}IR$Hv)Xq{zXdN|0m!7r{LNCgohZI4+MY!5C8%|00;m9AOHk_
z01yBIpLYUe{~w<JKkuUq_5lGP00e*l5C8%|00;m9AOHk_z)2?n`~N511HgVD00e*l
z5C8%|00;m9AOHk_01!9{1YrOFB#biH2n2ut5C8%|00;m9AOHk_01yBIC!GN7|DSXZ
z0Q-Re5C8%|00;m9AOHk_01yBIK;R@0fc^iIFv?&f5C8%|00;m9AOHk_01yBIKmZ7w
zbOP}Ff6_ew><0os00;m9AOHk_01yBIKmZ5;fs;T0KL39bMj31b0zd!=00AHX1b_e#
z00KY&2mpbTPJlfBhk^{nW+ML*8P0ukZhrQ9=;N7>roSI#0ymkj`+nxD`rbL|l)(N^
zL%{!s(c3(Ic(c=!(ZkxoZmZv2L+wtlqjdVcruMeCv)<{VlH4qL39-ehP^b%Bsj?>A
z;kJ*|Mk5^J@N1*umTz#&+xBi7vBSl1Fj~ZFUiDV9BJLt@<suW5Giy#zi3m2h;zqel
zY^`h*3pFA3)wkz@QJ$wi<Q1{iMNZS5=V4u$df`f0@UT1_TGqI@AeQ6?=35u83RP~i
zTDnoF-sP?fce%n=eWO&t&E60yb*vTs+bWkY>aA=J`VTrX8dz%^GMWe(gsQMERE0`W
zsBzmy4WXxWl!M0d=mKscW>&%DwzauNKwu2T!kS<!S`*5GS!5`q0j%6AREyUN)#XGw
z9UCd^;A&J^-|u!0ZtjcSPP>C-uLkVnVbr^hy~N~YSZtIQBnb_NomQtyOdMf#iaC7H
zC1;v!1xNy=b}&@X?g*jt5<$tCwU5KH+?PZWpVj{R;|(x`iv8~XZqF?j4#f^H%?6`Q
z99K;>u2fAWvDZU`qA1-%O}FY=lsN_G>`~?{9@~y3=A3n~@40(##2#;j)QJ0_p+=mu
zmh*hp9&yfM+`kcV&LV5f93kTL0rk?^<vF&9Q~RClFwf>MMp#>oGW^}9jUhO4UgLr*
z{=I{|ez1qLq>-tNE{w>lcXrWG5qI|*97!Ja(hZzRalRs+$j=0$mtyoIR`o=qThd;W
z9E=N^Vj+E4w<@KZTcbOVAv3CEaWA7xHw@onj4PVTjzlsly@dut9B8i0|A?9nMq@Gh
z@nzKyjmEU*?PoJ~AG8-*JkThx4$=`kPxFkC!kRf|nq8fkWsQDErXHm-8L}@2gHeGu
zhRo^6%}sHDdP<X|ZD-lE=lD6LtSwlE(kVtJ3|@EVi}-e?s^F|ZeEIN7AQ*iQXTAe7
z^I4)$@7i6{lK(8?LERpG3goOdp|uINJ;jK{DeJVz)h(got#Dw1!e-40%6Wm0J-+7G
zl628b(iU1yJ*Vse;w-j=h_mQ;p`z(*difD%B4fwE%ynuWBl9WpI>waX*)n`-NZ2&E
zD<$_duq+Ohjn)ti9w0m|GBd9}TjbV+^}<%U&Pn?NHDVPbP2(W%Gr?$li9S55hDi-6
zN~gD@#<V`))~@*N{^XWZ-AJA8YZb-?ZEVvQh+eor-&H+lcpuNvP;-dx@1q``wb@S@
zo8LGJmwVXxFXEc9!>e>Ky0k<;LTa13Zu&o;t&MH{s>S8>E6>f-Zi(G}JdO_U)NYdM
z^FQ?=#a8A%p8d(}#&jb1Yk|M;-(Y?hUwp2w4;EhuF7o{JN8*DHdY{bGH$~;1diU#~
zq4Qj4e5)kf=A31mI1Gnwm+IGW%oMk(wbCuYEmNrJ(=-xWHT-x{taI*Mz!hrT6>(^r
zf0yKo+_>N+@<q+(H4*LZO{$O-VJlYdaqHEM8?Ii>sKs*;?B4}4n;}|~m!_$CPOh!j
zR@|EF6C_ejURdVS>+I@Awc#sY;m-f+c@F>ETp8Jljj@`p+%xNHMmolmNN<`lrAh@q
zhQ5OzJY6`5<a!N1*PGIw26Jy)^&wfuV2^W7IAjcQZRM9PI`kfnydxZXnzBbx(t9ve
z-Q0eJz4RJ>PWJW~u2fTJvWLs3gVF66eR$U~tymKkaqMG|=@3NhReLVA7mjVi^FB?Q
zBbnB0!@%5#{l>#n>J0X47aTWfQpv19&KB)S(q3rEmG*+;rA)Jpk*elBR=;DJ&UK9Y
z>@z-k8?$Arb;fg!?Y3_SYGyobDdlO8Qs#YN@<jd^^BVnbQy1>oSB~o9U<WB)x1ljh
zHN}2UK@X2ogFHDT7Vxq|(zfiN&JU}ojc_uMP;>N-_=B<XX!^#e_?2<NGLrCa$}?O>
zb?|77-fIfrCx*oQB&FBT9_q<rPnn?f>B&laHZ@UcN3BNegV)($G?}D7+I9rF%{*<v
zYIJFST)2-UJY#-T>JHpb{}}5PUfyAU^f>#}OO{TY|G$aIH0L-=fFs(vf10|z*as1t
zUSv|(;6~YHOz!v$lg()5$-XY7uQ|OL3Bu+_iR|fEGK+K5V(gO|Y<|Gz$FWvAo^b|d
z>}#C7%@6qg|8rgBfX;vb5C8%|00;m9AOHk_01yBIKmZ855CZW0|3a`sr$7J*00AHX
z1b_e#00KY&2mk>f00f>Z0r>sD=ZYIT0|Gz*2mk>f00e*l5C8%|00;m9An-y6!2bUW
g!491Q0U!VbfB+Bx0zd!=00AHX1b_e#c&-Hg4<jA8Y5)KL

literal 0
HcmV?d00001

diff --git a/__init__.py b/__init__.py
index 1befa31..d83f4fb 100644
--- a/__init__.py
+++ b/__init__.py
@@ -10,6 +10,7 @@
 
 import math
 import torch
+import numpy as np # <-- Required for robust slicing of PyTorch tensors
 
 try:
     from diffusers.schedulers.scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler
@@ -23,6 +24,8 @@ from comfy.samplers import SchedulerHandler, SCHEDULER_HANDLERS, SCHEDULER_NAMES
 
 # Default config for registering in ComfyUI
 default_config = {
+    "start_at_step": 0,
+    "end_at_step": 9999,
     "base_image_seq_len": 256,
     "base_shift": math.log(3),
     "invert_sigmas": False,
@@ -60,7 +63,19 @@ class FlowMatchEulerSchedulerNode:
                     "default": 9, 
                     "min": 1, 
                     "max": 10000,
-                    "tooltip": "Number of diffusion steps. Z-Image-Turbo uses 9 steps by default (8 DiT forwards). Higher = better quality but slower."
+                    "tooltip": "Total number of diffusion steps to generate the full sigma schedule."
+                }),
+                "start_at_step": ("INT", { # <-- NEW INPUT
+                    "default": 0,
+                    "min": 0,
+                    "max": 10000,
+                    "tooltip": "The starting step (index) of the sigma schedule to use. Set to 0 to start at the beginning (first step)."
+                }),
+                "end_at_step": ("INT", { # <-- NEW INPUT
+                    "default": 99999,
+                    "min": 0,
+                    "max": 10000,
+                    "tooltip": "The ending step (index) of the sigma schedule to use. Set higher than 'steps' to use all steps."
                 }),
                 "base_image_seq_len": ("INT", {
                     "default": 256,
@@ -125,11 +140,13 @@ class FlowMatchEulerSchedulerNode:
     RETURN_NAMES = ("sigmas",)
     FUNCTION = "create"
     CATEGORY = "sampling/schedulers"
-    DESCRIPTION = "FlowMatch Euler Discrete Scheduler with full parameter control. Outputs SIGMAS for use with SamplerCustom. Supports Karras sigmas, dynamic shifting, and stochastic sampling for advanced control over the diffusion process."
+    DESCRIPTION = "FlowMatch Euler Discrete Scheduler with full parameter control and ability to trim the schedule (start_at_step/end_at_step)."
 
     def create(
         self,
         steps,
+        start_at_step,  # <-- New parameter
+        end_at_step,    # <-- New parameter
         base_image_seq_len,
         base_shift,
         invert_sigmas,
@@ -165,11 +182,24 @@ class FlowMatchEulerSchedulerNode:
 
         scheduler = FlowMatchEulerDiscreteScheduler.from_config(config)
         
-        # Set timesteps and get sigmas for the specified number of steps
+        # 1. Generate the full sigma schedule
         scheduler.set_timesteps(steps, device="cpu", mu=0.0)
         sigmas = scheduler.sigmas
+        
+        # 2. Apply start_at_step and end_at_step (Slicing the sigmas tensor)
+        # Determine the exclusive end index for the slice
+        # end_at_step is the step index (e.g., 5). We use 5+1=6 for the slice end index.
+        end_index = min(end_at_step + 1, len(sigmas))
 
-        return (sigmas,)
+        # Slice the tensor: [start:end]
+        sigmas_sliced = sigmas[start_at_step:end_index]
+        
+        # Check for empty schedule resulting from slicing
+        if sigmas_sliced.numel() == 0:
+            print("Warning: start_at_step/end_at_step resulted in an empty sigma schedule. Using full schedule as fallback.")
+            sigmas_sliced = sigmas
+            
+        return (sigmas_sliced,)
 
 
 # Import Flash Attention node
@@ -184,4 +214,4 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
     "FlowMatchEulerDiscreteScheduler (Custom)": "FlowMatch Euler Discrete Scheduler (Custom)",
     **FLASH_ATTN_DISPLAY_MAPPINGS
-}
+}
\ No newline at end of file