mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
fix
This commit is contained in:
@@ -73,10 +73,10 @@ _MODULE_NAME_TO_ATTRIBUTE_MAP_FLUX = {"x_embedder": "in_channels"}
|
||||
|
||||
def _maybe_dequantize_weight_for_expanded_lora(model, module):
|
||||
if is_bitsandbytes_available():
|
||||
from ..quantizers.bitsandbytes import dequantize_bnb_weight
|
||||
from ...quantizers.bitsandbytes import dequantize_bnb_weight
|
||||
|
||||
if is_gguf_available():
|
||||
from ..quantizers.gguf.utils import dequantize_gguf_tensor
|
||||
from ...quantizers.gguf.utils import dequantize_gguf_tensor
|
||||
|
||||
is_bnb_4bit_quantized = module.weight.__class__.__name__ == "Params4bit"
|
||||
is_gguf_quantized = module.weight.__class__.__name__ == "GGUFParameter"
|
||||
|
||||
@@ -14,12 +14,12 @@
|
||||
import copy
|
||||
from typing import TYPE_CHECKING, Dict, List, Union
|
||||
|
||||
from ..utils import logging
|
||||
from ...utils import logging
|
||||
|
||||
|
||||
if TYPE_CHECKING:
|
||||
# import here to avoid circular imports
|
||||
from ..models import UNet2DConditionModel
|
||||
from ...models import UNet2DConditionModel
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
|
||||
Reference in New Issue
Block a user