We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6cf941c commit fdf1c11Copy full SHA for fdf1c11
src/diffusers/models/model_loading_utils.py
@@ -54,6 +54,7 @@
54
}
55
56
57
+
58
if is_accelerate_available():
59
from accelerate import infer_auto_device_map
60
from accelerate.utils import get_balanced_memory, get_max_memory, offload_weight, set_module_tensor_to_device
0 commit comments