From 2ccd267404e59b4cee8de08877d6680a063c9854 Mon Sep 17 00:00:00 2001
From: nofunstudio <jimmy@nofun.io>
Date: Wed, 9 Apr 2025 11:17:37 -0600
Subject: [PATCH] more fixes for comfydeploy

---
 __pycache__/ksampler_fun.cpython-312.pyc | Bin 9348 -> 9699 bytes
 ksampler_fun.py                          |   9 ++++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/__pycache__/ksampler_fun.cpython-312.pyc b/__pycache__/ksampler_fun.cpython-312.pyc
index 71d151d3cd22f761aa470cdb77285ef3b1223123..e816d6a71056c409fa90d9e39cd9d220302dbf9f 100644
GIT binary patch
delta 1428
zcmZ{kZ)jUp6u{qo`Mcz$c}bc!Y0@ld`tPk<+HKl0yR~*5X|rljD$WX;#pKbZOPhLM
zvVUf4i!E-T(t2c_;IP6(WQdh014ZzYeo#iHg^hsqgJa?cK^L?#zj)8f;<V_2`@1Li
zo_p^3b6*xBKaA9TWwDqzCPP<#A0KG?ux5!T-xS{B3%oY(nhe5GQv*Dq`-IrRs(;5)
zGLOsI7?qV=I;#{u(w`!xhuGOVHk8*2J;p_{({;nva^2SQt*xzSZZB%v%ji6Ts0yhl
zr0j!@ZEjOiv%o#AJEfI$@aHxUc>wB7Z;0buo{On*p>ehzHcVc)W^#kg?0~x_bFilB
zEzeDQtNQ(UPRMSDcJmp#NJDIub|X3wZE)2*W;EQ(0jxq~`&J8Prsx>_D0Cm35O$Im
z;j$1h+^lz>)4nXg&qB{^FSAJ8%Y~^Nr>P85Zeq#`mn2D^1&2FZBt=qWl#?{k5}!+z
z^r@-%xU5Ko4lz46J<28p9kaU`X)D8N&tp_Ro{~=t#?$HN;)!FSp?`GZFrL6B%yBox
z+NG1LweNc0_7>|xMQdl#*h$q;F{7#&nZm*)WxW<2wgmXwdAQ0;%byzyuUon`E*3Vn
zb&E_MfK^ScaHZNmF53@Zi<e0lp0{<7LvYC^8U2+?hv1s6o{Ye4TQlhdv%S|dz?_vF
z7_$;TF2~fIV>9vW1RaF9eed2d+oy*RSOpqDJi;K^)q2n)OfMOe@^mJp$jK5pNuPqx
z>>XqbezHr}XK>e$$mFK8GNsQ#z|lwc!BIz>UM=Pqg;yL)r1ck(vpH9UW#NVxye<Y;
z!|P(lEzwzYb*;JA%*DF>>*9gUQP}PDSauBcXuj;x4ReO5q=zwQ%Rm$(!dB217~bR5
zqE;G&{iX;knqp^I;O_7iU7XzU5nWeV-=*uLy@ijR=XtUS{jP}k5}r7R@F5lp7hMM8
z-Lk5ttfMTcB(wO`9p#<>VL=zc=UvdZp`SW<-`m)*3)MXc^}$<F>q6{?@4Zek55IX^
zSam#&tz@onu<<OZ#rGlc^i?!MFg;XQ_gx~s$~jyj6tk0RD($#`-Jbuyj@c_U4C;6E
zH`oWY;dhfa;SYbf8*h$E8oI=0x{Tl%B!T8pKZ7{SP%<adaYdPuvlE#ly#U$35%(-^
z;`vGw=21h;!mU80S>3>&r-g8C+CMTHpPEU_R5GYng`?3hU53Y-Qo&(nQxc{#NjWwl
zr)OlU&|~Oo0D=EoREF=Gc8Ld39Yo->q-kgfHkh#_YUk-Z^amdd{yV0-#{E;7WG*fD
j(GQp*VMF;5R)T9bX>*ti64Li8hul2O!;{TBv>5YW3xhwP

delta 1195
zcmZ{kO-vI(6o7ZOKQ!&Og|@U*ONEwV)>@Pp6afjKRx4WnBV2^qu7Ck*W~&xm2~kWm
zdceGpXreJD8h>Ljt6se5$%D~g;%O64#+wHf6vM$dZ<}f$I>~<XcHYmtdGmHZc8(_;
zukCg#$Hvb)Kl-|=A3G*^@=U$Kt31g-*7Tf|LDYO({a_v;*7eNH%O3f>+G{ySLZ0)@
z(?;(MdZ!V^K?(wysem%8ziPxF8fl2FrA<W(HXGSv;5af<3%jgWg<N@2cf<?3OFiJT
z`GB{%>mKk0g~IjNizW(pWJyu}JC6yuO32zyyPPz}R%rvG4p9rgY~9c+Y+bfk!L%}F
zRfU5#!Yg4VQD9a`z$W{;Koip}NZP^{^rWW^(C(<_Un@{w;e|K6`|=BmI_X$z;B(Dj
zaNQ!C;jSx6Ho|+CSQRX!qVFAH5hy0Fl?-Ql`e-Uc6Hw`{A$8E<ZYFIo>h_ZkxZ++T
zZbs9Mh;0m_i*};Che5Msq~Yv<EM+v3r+eX-J4*J!3XkaQLTyPpdu%u-QQ8fMJguY=
zZhDrR6HFsNQQvwdNXx8)EAviQte)~u*`~cM?@Kq#?I4`1dSclZJ10cv1YMq&vEVNa
zap#tR;Z?x2B~p?UP4LhgS+|`j=y8U79K|BhMg-om4qVcm$GLC3U`mmhK+M!in^c$Y
z0#C-^nm<u`293`mf`~EoyT6147Om=sHBok)46~?Mn&iv>VL`{>ROPs>7UN)l(&g~E
zx-wdiqJIBQlvW{DGl(YYW0Pi)Q$t4wB`Q+|_5>rWPWfOh8BuQpM~MfIAoBDAIw6=G
zQ|qg)lknd-gyI(=ItmxFX#3dzwwE2uN$<BmV2#%t@{>u}ABwNVos*)0USu=9gy0!O
zfu2A)j~HRlZ0W(2EDuS!{%nR`hIgR@{*$Og|H6GJC?WFD9!{8bg#r^{j}>FblgT)}
z1dqZ4;W&HMgyC#P>gksTk77v<qi-AHAfgwVYGMuTD0U#QBWWLlW|r~;a!%IFsdQSB
zWzBe?XJ@Bgm_k1&aIK~xyfCwQOUqC;b8Jv*rT1A3{9nkQV76w;>6_b1T1e|xj$uwA
L@HDc*h<X18zuyI!

diff --git a/ksampler_fun.py b/ksampler_fun.py
index 2f0a347..da1ef40 100644
--- a/ksampler_fun.py
+++ b/ksampler_fun.py
@@ -53,7 +53,7 @@ def fun_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, neg
     
     out = latent.copy()
     out["samples"] = samples
-    return (out, None)  # Return None for preview_latents since we're saving directly
+    return (out, callback.get_results())  # Return the results from the callback
 
 class PreviewCallback:
     def __init__(self, latent, preview_steps_list, vae, model):
@@ -65,11 +65,15 @@ class PreviewCallback:
         self.type = "output"
         self.prefix_append = ""
         self.compress_level = 4
+        self.results = []  # Store results for later retrieval
         # Initialize previewer using the model's latent_format
         self.previewer = latent_preview.get_previewer("cpu", model.model.latent_format)
         if self.previewer is None:
             print("Warning: No preview method available, falling back to full VAE for previews")
 
+    def get_results(self):
+        return {"ui": {"images": self.results}}
+
     def __call__(self, step, denoised, x, total_steps):
         if step in self.preview_steps_list:
             # Create a copy of the current state
@@ -115,6 +119,9 @@ class PreviewCallback:
                     "output_id": "preview_images"
                 }
                 
+                # Add to results list
+                self.results.append(result)
+                
                 print(f"Saved preview image to: {file_path}")
 
 class KSampler: