3j5 SSKrSSKrSSKJs Jr SSKJr SSKr SSK r SSK r SSK r SSK JrJrJr SrSrSr/SQr/SQrS rS rS \R0S \R0S \S\S\R04 Sjr"SS\R65r"SS\R65r"SS\R65r"SS\R65r"SS\5r S\ 4Sjr!g)N)override)ComfyExtensionIOTypesi )g ףp= ?gv/?gCl?)gZd;O?gy&1?g?iiimagemask erode_radiussizereturnc ZUSSS24RSS5RSS5nURSS5Sn[R"XE/S5SnURSSupxU[ X5- n [ RRU[S[X-55[S[Xy-55SS5RSS5nUSS2SS 24n US:aA[R"U *S U-S-SUS 9*n [R"USS2SS24U /S5n[R"U S S:S S9upU R5S:Xa [S5e[!U R 55[!U R55p[!U R 55[!U R55nnX-S - UU-S - nn[X- UU- 5S - S-n[!UU- 5[!UU- 5[!UU-5[!UU-54unnnnURSSunnUR#SS UU- UU- 45n[US5[US5[ UU5[ UU54unnnnUU:a2UU:a,USS2SS2UU2UU24USS2SS2UU- UU- 2UU- UU- 24'[ RRUX3SS5RSS5nUSS2SS24USS2SS 24-SRSS5nUR%S5$)N.rlanczosdisabled)stridepadding)rrT)as_tuplez@TripoSplatPreprocessImage: mask is empty (no foreground pixels).g333333?)clampmovedimtorchcatshapemincomfyutilscommon_upscalemaxroundF max_pool2dnonzeronumel ValueErrorint new_zeros unsqueeze) r r r r rgbalphargbahwsaysxsx0x1y0y1cxcyhalfleftupperrightlowerHWcropsx0sy0sx1sy1outs =/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_triposplat.py _preprocessrJs RaR.  q! $ , ,R 3C JJq! T "E 99c\1 %d +D ::bc?DA s1yA ;; % %dC5<,@#aquBVXacm n t tuvxy zD Q!V Aa \\1"a,.21l S Syy$q"1"u+q)1- ]]1T7Q; 6FB xxzQ[\\ ]CM ]CMBg]R"WMB rwR 1 $s *D #BIBIBIPSTVY]T]P^ ^D% ::bc?DAq >>1a = >DT1s5!}c%mSPQ]RCc3 SyS3YEI!QPSTWPWY\]`Y`J`EaQ3;sU{*C$JsTz,A AB ;; % %dD : N T TUVXY ZD 2A2;a1f %q ) 1 1!R 8C == cP\rSrSr\S5r\S\R4Sj5rSr g)TripoSplatPreprocessImage>cf[R"SSSS[RRS5[RRS5[R RSSS S S S 9[R RS SSSS SS9/[RR SS9/S9$)NrMzTripoSplat Preprocess Imagemodel/conditioning/triposplatzPCrop center each image to a square canvas on a black background and add padding.r r r rrrzQErode the alpha matte by this pixel radius before cropping (avoids border bleed).defaultrr#tooltipr izZSquare image size. The model is trained at 1024; other sizes run but are off-distribution.rRrr#steprS display_namenode_idrYcategory descriptioninputsoutputs)rSchemaImageInputMaskIntOutputclss rI define_schema'TripoSplatPreprocessImage.define_schema?syy/64jw' f% ^QA2%xz VTs2&BC XX__'_:;  rKr c [S[U5S-S-5nURSURS:wa,[RR X!RS5n[ URSS5[ URSS5:waI[R"USS2S4R5[ URSS5SSS9SS2S4n[R"[URS5Vs/sHn[XX%X45PM snSS9n[R"U5$s snf) NrrrrbilinearF)r mode align_corners)dim)r#r*rr r!repeat_to_batch_sizetupler% interpolatefloatrrrangerJr NodeOutput)rgr r r r iprepareds rIexecute!TripoSplatPreprocessImage.executeQs2D R2-. ::a=EKKN *;;33D++a.ID AB E%++a*:$; ;==ag!4!4!6U5;;qQRCS=T[euz{|}@A}ABD99Z_`e`k`klm`nZopZoUVk%(DG\PZopvwx}}X&&qs EN __name__ __module__ __qualname____firstlineno__ classmethodrhrrtrw__static_attributes__ryrKrIrMrM>s3  "'''rKrMcP\rSrSr\S5r\S\R4Sj5rSr g)TripoSplatConditioning\c[R"SSSS[RRSSS9[RRSS S9[R RS 5/[R RS S 9[R RS S 9[RRSSS9/S9$)NrzTripoSplat ConditioningrPzEncode the image with DINOv3 and the Flux2 VAE into TripoSplat positive/negative conditioning, and create the fixed size noise target (latent + camera) for the KSampler clip_visionzDINOv3 ViT-H/16+ image encoderrSvaez Flux2 VAEr positiverXnegativelatentz-The fixed size noise target (latent +camera).)rYrSrZ) rr` ClipVisionrbVaera ConditioningreLatentrfs rIrh$TripoSplatConditioning.define_schema]syy,24r ##M;[#\ UK 8w' &&J&?&&J&?   h@o p  rKr c[RRUR5 URnUR SS5R U5n[R"[US9RSSSS5n[R"[US9RSSSS5nXV- U- nURUR5S9Sn[R"UR5UR SS5R [RR#55n UR%U5R [RR#55n U R Sn U SU /0//n [R&"U 5S[R&"U 5/0//n [RR#5n[R("U [*[,/US9n[R("U S[./US9n[R0R3UU45n[4R6"XSU05$) Nrrdevicer) pixel_valuesrreference_latentssamples)r model_managementload_model_gpupatcher load_devicertortensor _DINOV3_MEANview _DINOV3_STDmodelrrr% layer_normrintermediate_deviceencode zeros_likezeros_Q_TOKEN_LENGTH_LATENT_CHANNELS _CAM_CHANNELS nested_tensor NestedTensorrrt)rgrrr rimgmeanstdseqfeature1refbrrdev latent_seqcamerars rIrwTripoSplatConditioning.executeqs --k.A.AB((mmB"%%f-||L8==aAqIll;v6;;Aq!QGzS SYY[9!<<< SYYrs^<??@V@V@j@j@lmjj""5#9#9#M#M#OP IIaL 3cU;<=%%h/2EHXHXY\H]G^1_`a$$88:[[!_6F!GPST aM23?%%22J3GH}}X)W1EFFrKryNrzryrKrIrr\s6  &GGGrKrcP\rSrSr\S5r\S\R4Sj5rSr g)VAEDecodeTripoSplatct[R"SSSS[RRS5[RRSSS9[R RS S [ [S S S 9[R RSSSSSS9/[RRSS9/S9$)NrzTripoSplat Decode 3d/latentzrDecode the sampled TripoSplat latent into a 3D gaussian splat. Modify the number of gaussians to vary the density.rrTripoSplat VAE decoderr num_gaussians zNumber of gaussians to produce (rounded to a multiple of 32). 262144 matches the octree's point density; higher oversamples the same points (denser, but no new detail) and costs proportionally more VRAM/time.rVseedrlzFSeeds the octree point sampler (global RNG) for deterministic decodes.rQsplatrXrZ) rr`rrbrrd_NUM_GAUSSIANS_MIN_NUM_GAUSSIANS_MAXSplatrerfs rIrh!VAEDecodeTripoSplat.define_schemasyy), N  * U,D E _fBTZlsu&lm VQA;M%mo XX__'_:;!  rKr c USn[USS5(aUR5SOUnURnURn[ [ [ [[U555n X-S:wa[X- 5U-n [RRUR5n URRn UR Sn U S-X-S--U -U -n [RR#UR$/U S9 UR'UR(URS 9n[*R,"S S 9R/U5nUR1XiUS 9Vs/sHoR35PM nnS [5U65unnnnn[6R8"[:R<"UUUUU55$s snf)Nr is_nestedFrrr memory_required)rdtypecpur)r generatorc3N# UHn[R"U5v M g7fN)rstack).0ts rI .VAEDecodeTripoSplat.execute..s6[{!u{{1~~{s#%)getattrunbindfirst_stage_modelgaussians_per_pointr#rrrr*r$r r dtype_size vae_dtypegsmodel_channelsrload_models_gpurrrr Generator manual_seeddecoderender_tensorsziprrtrSPLAT)rgrrrrr2rdecodergppnrhidden cond_tokensrrgparts positionsscales rotations opacitiesshs rIrwVAEDecodeTripoSplat.executes I ")![%"@"@Aa'')) "C(:C Octree depth for the preview decode (lower = cheaper/coarser).)rRrr#advancedrSri@rTrrzMNumber of gaussians to produce for the preview (rounded to a multiple of 32).rVyawgV@gvgv@g?zPreview camera yaw in degrees.)rRrr#rWrSrpitchg.@g@Vg@V@z Preview camera pitch in degrees. point_sizerrrzMaximum splat radius in pixels. Each gaussian is sized from its scale and capped here; lower = finer/pointier, higher = chunkier.rQrZ)rr`ModelrbrrdFloatrerfs rIrh'TripoSplatSamplingPreview.define_schemasyy/6 ?w' U,D E ^QA1t%eg _e6XZ%tvudEPS]}IMOw%TPS]KOQ \1!&RS XX__&''  rKr c^^ ^ ^SSKJm XCXVUS.m TRnURRR n U S-XHR -S--URR-[RRTR5-mU U UU4Sjn UR5n U R[RR R"SU 5 [$R&"U 5$)Nr)decode_x0_to_image) gaussianslevelrrrrrc>^^[U5nSn[U5U:aXOURS5mSSSS.mUUU UUU 4Sjn[U5U:aXAU'OXBS'U"U0UD6$)NrcallbackTF)okpbarloadedc>T b T "XX#5 T S(dgT S(d[[RRSS9nURT R5 [RR UT S9 ST S'T"T UT5nT Sc"[R RU5T S'T SRUS-USUS 45 g![a4n[R"S RU55 S T S'SnAgSnAff=f) NrrT)only_currently_usedrrrJPEGiz9TripoSplatSamplingPreview: preview failed, disabling ({})F) r r loaded_modelsappendrrr! ProgressBarupdate_absolute Exceptionloggingwarningformat) rWr6x total_stepsrrecfgrrorig_cbstaters rIrQTripoSplatSamplingPreview.execute..outer_sample_wrapper..callbacks&Da5T{ ( ?(-(>(>(L(Lae(L(f %,,S[[9..>>}^m>n*.h,S"c:CV},(- (?(? (Lf &M11$(K&RUWZI[\ (OO$_$f$fgh$ij"'E$K(sB3C D *DD )listlenget) executorargskwargscb_idxrrrrrrrs @@rIouter_sample_wrapper?TripoSplatSamplingPreview.execute..outer_sample_wrappersq:DF&)$i&&8dlfjj>TG?E ( (&4y6!'V %-z"T,V, ,rKtriposplat_sampling_preview)comfy.ldm.triposplat.previewrrrdiffusion_modelq_token_lengthrrrr rrrcloneadd_wrapper_with_keypatcher_extension WrappersMP OUTER_SAMPLErrt)rgrrrrrrrfsmrrmrrrs ` @@@rIrw!TripoSplatSamplingPreview.executesC)')##kk11@@ &?m?V?V.VZ\-\\`c`f`f`u`uux}yOyOyZyZ[^[h[hyii - -> KKM u66AANNPmpD E}}QrKryNrzryrKrIrrs6  ., Y[YfYf, , rKrcL\rSrSr\S\\\R4Sj5r Sr g)TripoSplatExtensionir c6# [[[[/$7fr)rMrrr)selfs rI get_node_list!TripoSplatExtension.get_node_lists & "  %   sryN) r{r|r}r~rrtyper ComfyNoder0rryrKrIr-r-s)  T$r||*<%=  rKr-c# [5$7fr)r-ryrKrIcomfy_entrypointr5 s   s )"r rtorch.nn.functionalnn functionalr%typing_extensionsrcomfy.model_managementr comfy.nested_tensorcomfy.patcher_extension comfy.utilscomfy_api.latestrrrrrrrrrrTensorr*rJr3rMrrrr-r5ryrKrIr@s &66 $ # "u||"5<<"s"RU"Z_ZfZf"J' '<-GR\\-G`*W",,*WZF  F R . ! 3!rK