Comments fix

Co-Authored-By: Ryan Dick <[email protected]>

Author: StAlKeR7779

invokeai/backend/stable_diffusion/extensions/base.py

@@ -65,7 +65,7 @@ def patch_unet(self, unet: UNet2DConditionModel, original_weights: Dict[str, tor
 
         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.
         """

invokeai/backend/stable_diffusion/extensions/lora.py

@@ -57,7 +57,7 @@ def patch_model(
         :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"