From 2a3483d2b0dce6edc6544d0072a6ce3b1eb550db Mon Sep 17 00:00:00 2001
From: db0 <mail@dbzer0.com>
Date: Sat, 9 Dec 2023 17:20:44 +0100
Subject: [PATCH] fix: must initiate final_rawpng

---
 hordelib/horde.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/hordelib/horde.py b/hordelib/horde.py
index 343ec6f5..ae472f00 100644
--- 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(