diff --git a/api/tests/test_diffusers/test_load.py b/api/tests/test_diffusers/test_load.py index 22ca1a8d..286c166d 100644 --- a/api/tests/test_diffusers/test_load.py +++ b/api/tests/test_diffusers/test_load.py @@ -1,6 +1,6 @@ import unittest -from unittest.mock import MagicMock from os import path +from unittest.mock import MagicMock import torch from diffusers import DDIMScheduler, OnnxRuntimeModel