more clean-up

src/compressed_tensors/quantization/lifecycle/forward.py

@@ -17,8 +17,6 @@
 from typing import Callable, Optional
 
 import torch
-
-# from compressed_tensors.quantization.cache import QuantizedKVParameterCache
 from compressed_tensors.quantization.quant_args import (
     QuantizationArgs,
     QuantizationStrategy,

src/compressed_tensors/quantization/utils/helpers.py

@@ -225,7 +225,7 @@ def iter_named_leaf_modules(model: Module) -> Generator[Tuple[str, Module], None
     """
     for name, submodule in model.named_modules():
         children = list(submodule.children())
-        # would the observer ever be attached when this function is called?
+        # TODO: verify if an observer would ever be attached in this case/remove check
         if len(children) == 0 and "observer" in name:
             yield name, submodule
         else:
@@ -257,9 +257,9 @@ def iter_named_quantizable_modules(
     :returns: generator tuple of (name, submodule)
     """
     for name, submodule in model.named_modules():
+        # TODO: verify if an observer would ever be attached in this case/remove check
         if include_children:
             children = list(submodule.children())
-            # would the observer ever be attached when this function is called?
             if len(children) == 0 and "observer" not in name:
                 yield name, submodule
             else: