From 8a948b80ee49f51b2ad9a6ca89bd9f42fdcb9915 Mon Sep 17 00:00:00 2001 From: Dowon Date: Tue, 16 Apr 2024 21:22:44 +0900 Subject: [PATCH] fix: python <= 3.9 union error --- aaaaaa/helper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aaaaaa/helper.py b/aaaaaa/helper.py index 2bcda32..27dd6a9 100644 --- a/aaaaaa/helper.py +++ b/aaaaaa/helper.py @@ -2,7 +2,7 @@ from __future__ import annotations from contextlib import contextmanager from copy import copy -from typing import TYPE_CHECKING, Any +from typing import TYPE_CHECKING, Any, Union import torch @@ -21,7 +21,7 @@ else: StableDiffusionProcessingTxt2Img, ) -PT = StableDiffusionProcessingTxt2Img | StableDiffusionProcessingImg2Img +PT = Union[StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img] @contextmanager