cleanup

modules/sdnq/__init__.py

@@ -3,6 +3,7 @@
 from typing import Any, Dict, List, Tuple, Optional, Union
 from dataclasses import dataclass
 from enum import Enum
+
 import torch
 from diffusers.quantizers.base import DiffusersQuantizer
 from diffusers.quantizers.quantization_config import QuantizationConfigMixin

modules/sdnq/forward.py

@@ -2,8 +2,6 @@
 
 from typing import Callable
 
-import torch
-
 from .common import conv_types, conv_transpose_types