remove duplicate imports

This commit is contained in:
Seunghoon Lee
2024-07-02 22:40:08 +09:00
parent cbd8b57c7c
commit 803558991d
-1
View File
@@ -34,7 +34,6 @@ class PipelineBase(TorchCompatibleModule, diffusers.DiffusionPipeline, metaclass
self.model_type = self.__class__.__name__
def to(self, *args, **kwargs):
import optimum.onnxruntime
if self.__class__ == OnnxRawPipeline: # cannot move pipeline which is not preprocessed.
return self