3jn,SSKJr SSKrSSKrSSKrSSKrSSKJrJ r "SS\ R5r "SS\ R5r "SS \ R5r "S S \5rS \4S jrg))overrideN)ComfyExtensionioc \rSrSr\S\R 4Sj5r\SS.S\S\S\S\R4S jj5r S r g ) EmptyHiDreamO1LatentImage returnc6[R"SSSS[RRSSSSS S 9[RRS SSSS S 9[RRS S S SS9/[R"5R 5/S9$)NrzEmpty HiDream-O1 Latent Imagezmodel/latent/hidreama,Empty pixel-space latent for HiDream-O1-Image. The model was trained at ~4 megapixels; lower resolutions go off-distribution and quality regresses noticeably. Trained resolutions: 2048x2048, 2304x1728, 1728x2304, 2560x1440, 1440x2560, 2496x1664, 1664x2496, 3104x1312, 1312x3104, 2304x1792, 1792x2304.widthi@i )iddefaultminmaxstepheight batch_size)rrrrnode_id display_namecategory descriptioninputsoutputs)rSchemaIntInputLatentOutputclss =/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_hidream_o1.py define_schema'EmptyHiDreamO1LatentImage.define_schema syy/8+T 24b Q $BDr R aQB G YY['')*!  r)rr rrc[R"USX!4[RR 5S9n[ R "SU05$)N)devicesamples)torchzeroscomfymodel_managementintermediate_devicer NodeOutput)r#r rrlatents r$execute!EmptyHiDreamO1LatentImage.execute!sB F *))==? }}i011r'N) __name__ __module__ __qualname____firstlineno__ classmethodrrr%intr1r3__static_attributes__r5r'r$rr sV bii  (CD2s2C2S222r'rc\rSrSrSr\S\R4Sj5r\S\RRS\R4Sj5r Sr g) HiDreamO1ReferenceImages*zCAttach reference images to both positive and negative conditioning.r c [R"SSSS[RRSS9[RRSS9[RRS[RR [R RS 5[S S 5Vs/sHnS U3PM snS S 9SS9/[RRSS9[RRSS9/S9$s snf)Nr>zHiDream-O1 Reference Imageszmodel/conditioning/hidreamzuAttach 1-10 reference images to conditioning, one for edit instructionor multiple for subject-driven personalization.positivernegativeimagesimagerrimage_)namesrzLReference images. 1 image = instruction edit; 2-10 images = multi reference.)templatetooltip)rr) rr ConditioningrAutogrow TemplateNamesImageranger!)r#is r$r%&HiDreamO1ReferenceImages.define_schema-syy.61B%%%4%%%4 !![[66w/5:1b\B\s|\B7 l"  &&J&?&&J&?-  Cs,D rDc[SS5Vs/sHnSU3U;dMUSU3PM nn[R"USU0SS9n[R"USU0SS9n[R"X5$s snf)NrrrFreference_latentsT)append)rN node_helpersconditioning_set_valuesrr1)r#rArCrDrOrefss r$r3 HiDreamO1ReferenceImages.executeKs.3ArlUlqclf>T$s|$lU77CVX\B]fjk77CVX\B]fjk}}X00Vs A4 A4r5N)r6r7r8r9__doc__r:rrr%rKTyper1r3r<r5r'r$r>r>*sRM bii  :1BKK4D4D111r'r>c \rSrSrSrSrSS//SQ/SQSS //S Q/S QS .rS /S/S S//SQS.r\SS\ S\ S\ S\ R4Sjj5r \ S\R4Sj5r\ S\S\S\S\S\S\S\R&4Sj5rSrg) HiDreamO1PatchSeamSmoothingSr rrr`)r^r`rrr`r_)r^rarbr_rd)rd)rdre)rerergrc)rf)rdrg)rgrdrc))rh)r]rh)rhr])r]r])ri) ri)rirj)rjrj)) single_shift)rkr])rkrd) symmetricrl)rmr])rmrdrl)rlr]rd24ramp_2_4 ramp_2_4_8cycxsizer cUS-n[R"U5RUS5n[R"U5RSU5nX@- U-U-U- nXQ- U-U-U- nSS[R"[RU-U- 5--S[R"[RU-U- 5--$)z9size x size Hann tile peaking at (cy, cx) within a patch.rlrg?)r,arangeviewcospi)rsrtruhalfyyxxdydxs r$ _hann_tile&HiDreamO1PatchSeamSmoothing._hann_tileisqy \\$  $ $T1 - \\$  $ $Q -w~% -w~% -q599UXX]T%9::;q599UXXXZ]]aMaCb?bccr'c4[R"SSSSS[RRSS9[RRSS S S S S S9[RRSS S S S SS9[R RSSS/SSS9[R RS/SQSSS9[R RS/SQSSS9[RRSS S S S SS9/[RR 5/S 9$)!Nr[zHiDream-O1 Patch Seam Smoothingzmodel/patch/hidreamTzyAverage the model output across multiple shifted patch-grid positions during the late portion of sampling. Cancels seams.modelrB start_percentg??g{Gz?z?Sampling progress (0=start, 1=end) at which the blend turns ON.)rrrrrrI end_percentz/Sampling progress at which the blend turns OFF.patternrkrmzShift layout. single_shift: one pass at the natural patch grid + others offset. symmetric: all passes off-grid, shifts split around origin.)roptionsrrIpassesrnrozNumber of passes per gated step. 2/4 = fixed. ramp_*: pass count increases as sampling approaches end (more smoothing where seams are most visible).blend)averagewindowmedianrzaverage: equal-weight mean. window: Hann-windowed weighting favoring each pass away from its patch boundaries. median: per-pixel median, rejects wraparound-outlier passes.strengthzLInterpolation between the natural-grid pred (0) and the averaged result (1).)rrris_experimentalrrr)rrModelrFloatCombor!r"s r$r%)HiDreamO1PatchSeamSmoothing.define_schemass6yy1:* P'*/3CSW[]-#3UYM +[9*j  @s  ;%J  *csRVj5<XX__&'O( ( r'rrrrrrc^^^^^^^^^^TS::dX2::a[R"U5$TRmTS-nTRUV s/sHn TRXI4PM sn mTS:Xa[TV V V s/sHHn [ R "U V V s/sH"upTRX- T-X- T-T5PM$ sn n SS9PMJ sn n n mOS/[T5-mUR5n U RS5n[UR5n[URU55U-m[URU55U-m0mS[S[S [ R4UUU4S jjmUUUUUUUU4S jnU R!["R$R&R(S U5 [R"U 5$s sn fs sn n fs sn n n f) Nrrlrrdimmodel_samplingHWr c ~>XX#4nTRU5nUbU$T Rn[R"[R"X[R S9US- [R"X[R S9- 5n[R"[R"X[R S9US- [R"X[R S9- 5nUT - U- R SS5n UT - U- R SS5n U SS2S4U SSS24-RU5n U TU'U $)Nr*dtyperr)get EDGE_FEATHERr,minimumrwfloat32clampto)rrr*rkeycachedfeatherysxsy_maskx_maskrampPr#edge_ramp_caches r$ get_edge_ramp:HiDreamO1PatchSeamSmoothing.execute..get_edge_ramps'C$((-F! &&Gu||AEMMR !Aaemm)\\^Bu||AEMMR !Aaemm)\\^BAv(//15FAv(//15F1d7OfT1Wo599%@D#'OC Kr'c >USn[USS5nU"U0UD6nTUs=::aT::dU$ U$[T5S:XaSnOBTU- [TT- S5- n[[ U[T5-5[T5S- 5nTUnT Un /n UHrupU S:XaU S:XaU R U5 M$[ R"X;U 4SS9n U"U /USSQ70UD6nU R [ R"X*U *4SS95 Mt [ R"U SS9nURu nnnTS:XaURSnU RURURS9nURSUT-UT-5SS2SU2SU24nURSS S 9n[ R"US :[ R "US U- 5UUR#SS 9- 5nUUSS2SSSS2SS24-RSS9nO5TS:Xa [ R$"USS9R&nOUR)SS9nT"UUURUR5nUS UT-- -UUT---$)Nrrg:0yE>))shiftsdimsrrrT)rkeepdimgMbP?r)rr)floatlenrrr;rSr,rollstackshaperr*rrepeatsumwhere full_likerrvaluesmean)!executorargskwargsxtpred level_idxphaser window_tilespredssysxx_rolled pred_rolledstacked_rrNtileswsum_wavgmaskrrend_tr shift_levelsstart_trwindow_tile_levelss! r$smoothing_wrapper>HiDreamO1PatchSeamSmoothing.execute..smoothing_wrappersQAd1gaj!AT,V,DQ)') * < A%  1GeOT(BBEC ,=$= >L@QTU@UV !),F-i8LE 7rQwLL& ::aRxH&xE$qr(EfE  UZZ S2#JXVW !kk%Q/G#MMMAq!Q MM!$$w~~W]]SLLAFAF3Arr2A2I>!T2KK eooaq.I1u{{_c{OdKde1dD!Q#6!77<<<C("ll7299llql)!At{{DJJ?D301C4(?4KK Kr'hidream_o1_patch_seam_smoothing)rr1 PATCH_SIZE RAMP_LEVELSSHIFTS_BY_PATTERNr,rrrcloneget_model_objectr multiplierpercent_to_sigmar;Tensoradd_wrapper_with_keyr.patcher_extension WrappersMPDIFFUSION_MODEL)r#rrrrrrrr{nlstrrmrrrrrrrrrrs` `` @@@@@@@r$r3#HiDreamO1PatchSeamSmoothing.executes s?k:==' ' NNAvEH__U[E\]E\--wl;E\] H ("'C _bc_bU[UWS^^TY!Odi1_aP_bcijk'"  #'#l*;!;  KKM++,<=>445 77 FG*Tn55kBCjP " S S ELL   ' L' LR u66AAQQStwH I}}QY^d"sG 7G)G; GGr5N)r )r6r7r8r9rrrr staticmethodr;r,rrr:rrr%rstrr1r3r<r5r'r$r[r[SsJL%h/AB&/AF cc!f Kdsdd3d dd) bii) ) VR eR %R RUR _bR knR zR EGERERR R r'r[cL\rSrSr\S\\\R4Sj5r Sr g)HiDreamO1Extensionr c,# [[[/$7fN)rr>r[)selfs r$ get_node_list HiDreamO1Extension.get_node_lists & $ '  sr5N) r6r7r8r9rlisttyper ComfyNoderr<r5r'r$rrs)  T$r||*<%=  r'rr c# [5$7fr)rr5r'r$comfy_entrypointrs  s )typing_extensionsrr,comfy.model_managementr.comfy.patcher_extensionrTcomfy_api.latestrrrrr>r[rrr5r'r$rsd& /2 2>&1r||&1R_ ",,_ D   2 r'