diff --git a/modules/sdnq/__init__.py b/modules/sdnq/__init__.py index e6c012e8c..ef3ef317f 100644 --- a/modules/sdnq/__init__.py +++ b/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 diff --git a/modules/sdnq/forward.py b/modules/sdnq/forward.py index 211b57768..ff2923dfb 100644 --- a/modules/sdnq/forward.py +++ b/modules/sdnq/forward.py @@ -2,8 +2,6 @@ from typing import Callable -import torch - from .common import conv_types, conv_transpose_types