Skip to content

Commit 7ad7cc8

Browse files
committed
try to fix partial import error
1 parent fa9a1f3 commit 7ad7cc8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@
3333
from torch import Tensor, nn
3434

3535
from .. import __version__
36-
from ..hooks import apply_layerwise_casting
3736
from ..quantizers import DiffusersAutoQuantizer, DiffusersQuantizer
3837
from ..quantizers.quantization_config import QuantizationMethod
3938
from ..utils import (
@@ -402,6 +401,7 @@ def enable_layerwise_casting(
402401
non_blocking (`bool`, *optional*, defaults to `False`):
403402
If `True`, the weight casting operations are non-blocking.
404403
"""
404+
from ..hooks import apply_layerwise_casting
405405

406406
user_provided_patterns = True
407407
if skip_modules_pattern is None:

0 commit comments

Comments
 (0)