Merge pull request #119 from Haidra-Org/main

fix: must initiate final_rawpng
Haidra-Org · Dec 9, 2023 · 8f875b4 · 8f875b4
2 parents 2b4f2d0 + af2d275
commit 8f875b4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/hordelib/horde.py b/hordelib/horde.py
@@ -11,7 +11,6 @@
 from collections.abc import Callable
 from copy import deepcopy
 
-import PIL.Image
 from horde_sdk.ai_horde_api.apimodels import ImageGenerateJobPopResponse
 from horde_sdk.ai_horde_api.apimodels.base import (
     GenMetadataEntry,
@@ -827,9 +826,9 @@ def basic_inference(self, payload: dict | ImageGenerateJobPopResponse) -> list[R
         if post_processing_requested is not None:
             post_processed = []
             for ret in return_list:
-                image = ret.image
                 single_image_faults = []
-                final_image: PIL.Image.Image | None = image
+                final_image = ret.image
+                final_rawpng = ret.rawpng
                 for post_processing in post_processing_requested:
                     if post_processing in KNOWN_UPSCALERS.__members__:
                         image_ret = self.image_upscale(