diff --git a/invokeai/backend/stable_diffusion/extensions/base.py b/invokeai/backend/stable_diffusion/extensions/base.py
index f9753b4344..61276e0784 100644
--- a/invokeai/backend/stable_diffusion/extensions/base.py
+++ b/invokeai/backend/stable_diffusion/extensions/base.py
@@ -65,7 +65,7 @@ class ExtensionBase:
 
         Args:
             unet (UNet2DConditionModel): The UNet model on execution device to patch.
-            cached_weights (Dict[str, torch.Tensor]]): A read-only copy of the model's original weights in CPU, for
+            original_weights (Dict[str, torch.Tensor]]): A read-only copy of the model's original weights in CPU, for
                 unpatching purposes. Extension can save tensor which being modified, if it is not saved yet, or can
                 access original weight value.
         """
diff --git a/invokeai/backend/stable_diffusion/extensions/lora.py b/invokeai/backend/stable_diffusion/extensions/lora.py
index 71584247c0..cfb97a2cb2 100644
--- a/invokeai/backend/stable_diffusion/extensions/lora.py
+++ b/invokeai/backend/stable_diffusion/extensions/lora.py
@@ -57,7 +57,7 @@ class LoRAExt(ExtensionBase):
         :param lora: LoRA model to patch in.
         :param lora_weight: LoRA patch weight.
         :param prefix: A string prefix that precedes keys used in the LoRAs weight layers.
-        :param original_weights: TODO:
+        :param original_weights: Dict of original weights, filled by weights which lora patches, used for unpatching.
         """
 
         # assert lora.device.type == "cpu"