hypertile autodetect optimal value

This commit is contained in:
Vladimir Mandic
2023-12-30 11:32:22 -05:00
parent c8b67d3197
commit c94d8ee4f2
4 changed files with 64 additions and 13 deletions
+2 -1
View File
@@ -1,3 +1,4 @@
from typing import Union
import time
from diffusers import StableDiffusionPipeline, StableDiffusionXLPipeline
from modules.control.proc.reference_sd15 import StableDiffusionReferencePipeline
@@ -13,7 +14,7 @@ def list_models():
class ReferencePipeline():
def __init__(self, pipeline: StableDiffusionXLPipeline | StableDiffusionPipeline, dtype = None):
def __init__(self, pipeline: Union[StableDiffusionXLPipeline, StableDiffusionPipeline], dtype = None):
t0 = time.time()
self.orig_pipeline = pipeline
self.pipeline = None