add TODO

neuralmagic · Apr 12, 2024 · 8465015 · 8465015
1 parent 9aae8e8
commit 8465015
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/sparsetensors/quantization/apply.py b/src/sparsetensors/quantization/apply.py
@@ -48,6 +48,7 @@ def apply_quantization_config(model: Module, config: QuantizationConfig):
         target = _find_first_name_or_class_match(name, submodule, target_to_scheme)
         if target is not None:
             # target matched, initialize layer from the matched scheme
+            # TODO: add follow on lifecycle calls based on the quantization status
             initialize_module_for_quantization(submodule, target_to_scheme[target])