Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

rework refiner

Browse Source 
rework refiner
This commit is contained in:
lllyasviel
2023-10-11 23:44:40 -07:00
committed by GitHub
parent 5e6b27a680
commit 132afcc2a2
12 changed files with 284 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/inpaint_worker.py
+7 -1
View File
@@ -167,6 +167,7 @@ class InpaintWorker:
# ending
self.latent = None
self.latent_after_swap = None
self.latent_mask = None
self.inpaint_head_feature = None
return
@@ -191,9 +192,14 @@ class InpaintWorker:
self.inpaint_head_feature = inpaint_head(feed)
return
def load_latent(self, latent, mask):
def load_latent(self, latent, mask, latent_after_swap=None):
self.latent = latent
self.latent_mask = mask
self.latent_after_swap = latent_after_swap
def swap(self):
if self.latent_after_swap is not None:
self.latent, self.latent_after_swap = self.latent_after_swap, self.latent
def color_correction(self, img):
fg = img.astype(np.float32)