Skip to content

Commit 735660d

Browse files
committed
remove todo
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 783a081 commit 735660d

File tree

1 file changed

+1
-1
lines changed
  • src/compressed_tensors/compressors/quantized_compressors

1 file changed

+1
-1
lines changed

src/compressed_tensors/compressors/quantized_compressors/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def compress(
102102
model_state[name] = value.to(save_device)
103103
continue
104104

105-
# compress values on cpu. TODO: experiment with different devices
105+
# compress values on cpu (memory movement too expensive)
106106
quant_args = names_to_scheme[prefix].weights
107107
compressed_values = self.compress_weight(
108108
weight=value,

0 commit comments

Comments
 (0)