Skip to content

Commit 1e4cb0b

Browse files
committed
reviewer feedback
1 parent 5cbf07e commit 1e4cb0b

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1550,7 +1550,6 @@ def _load_pretrained_model(
15501550
# tensors using their expected shape and not performing any initialization of the memory (empty data).
15511551
# When the actual device allocations happen, the allocator already has a pool of unused device memory
15521552
# that it can re-use for faster loading of the model.
1553-
# TODO: add support for warmup with hf_quantizer
15541553
if device_map is not None:
15551554
expanded_device_map = _expand_device_map(device_map, expected_keys)
15561555
_caching_allocator_warmup(model, expanded_device_map, dtype, hf_quantizer)

0 commit comments

Comments
 (0)