add optimization on gpu

This commit is contained in:
Seunghoon Lee
2023-10-29 20:51:02 +09:00
parent 150c3ad078
commit b7cfd09d50
7 changed files with 14 additions and 10 deletions
+1 -1
View File
@@ -26,7 +26,7 @@ def list_crossattention():
def get_pipelines():
import diffusers
from modules.olive import OnnxStableDiffusionPipeline, OlivePipeline
from modules.onnx import OnnxStableDiffusionPipeline, OlivePipeline
from installer import log
pipelines = { # note: not all pipelines can be used manually as they require prior pipeline next to decoder pipeline
'Autodetect': None,