fixes

Author: sayakpaul

tests/quantization/bnb/test_4bit.py

@@ -22,10 +22,9 @@
 import safetensors.torch
 
 from diffusers import BitsAndBytesConfig, DiffusionPipeline, FluxTransformer2DModel, SD3Transformer2DModel
-from diffusers.utils import logging
+from diffusers.utils import is_accelerate_version, logging
 from diffusers.utils.testing_utils import (
     CaptureLogger,
-    is_accelerate_version,
     is_bitsandbytes_available,
     is_torch_available,
     is_transformers_available,

tests/quantization/bnb/test_mixed_int8.py

@@ -20,9 +20,9 @@
 import pytest
 
 from diffusers import BitsAndBytesConfig, DiffusionPipeline, FluxTransformer2DModel, SD3Transformer2DModel, logging
+from diffusers.utils import is_accelerate_version
 from diffusers.utils.testing_utils import (
     CaptureLogger,
-    is_accelerate_version,
     is_bitsandbytes_available,
     is_torch_available,
     is_transformers_available,