From 393e36e3135bd054e62f29fb10f4695e5e26989e Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Thu, 25 Jan 2024 21:14:59 +0900
Subject: [PATCH] cleanup

---
 modules/sd_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 0d235b073..98c5af7ee 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -1090,7 +1090,7 @@ def set_diffuser_pipe(pipe, new_pipe_type):
     feature_extractor = getattr(pipe, "feature_extractor", None)
 
     # skip specific pipelines
-    if pipe.__class__.__name__ == 'StableDiffusionReferencePipeline' or pipe.__class__.__name__ == 'StableDiffusionAdapterPipeline':
+    if pipe.__class__.__name__ == 'StableDiffusionReferencePipeline' or pipe.__class__.__name__ == 'StableDiffusionAdapterPipeline' or 'Onnx' in pipe.__class__.__name__:
         return pipe
 
     try: