fix: python <= 3.9 union error

pull/595/head
Dowon 2024-04-16 21:22:44 +09:00
parent 6ca27572bc
commit 8a948b80ee
1 changed files with 2 additions and 2 deletions

View File

@ -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