PIL Image.Image type fix

2026-09-19 09:14:35 +02:00 · 2026-04-06 01:06:19 -07:00
parent a07e9cc7e1
commit c003b0eb48
15 changed files with 23 additions and 23 deletions
@@ -34,7 +34,7 @@ class DiffusionHeatMapHooker(AggregateHooker):
        locate_middle = load_heads or save_heads
        self.locator = UNetCrossAttentionLocator(restrict={0} if low_memory else None, locate_middle_block=locate_middle)
        self.last_prompt: str = ''
-        self.last_image: Image = None
+        self.last_image: Image.Image = None
        self.time_idx = 0
        self._gen_idx = 0