fix: python <= 3.9 union error

This commit is contained in:
Dowon
2024-04-16 21:22:44 +09:00
parent 6ca27572bc
commit 8a948b80ee

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