This commit is contained in:
Seunghoon Lee
2024-01-25 21:14:59 +09:00
parent 8bcd60065d
commit 393e36e313
+1 -1
View File
@@ -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: