Skip to content

Commit

Permalink
(minor) remove remaining incorrect references to T5 encoder in comments.
Browse files Browse the repository at this point in the history
  • Loading branch information
RyanJDick committed Sep 27, 2024
1 parent ca13c3b commit 536ccf0
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion invokeai/app/invocations/flux_lora_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ class FluxLoRALoaderOutput(BaseInvocationOutput):
classification=Classification.Prototype,
)
class FluxLoRALoaderInvocation(BaseInvocation):
"""Apply a LoRA model to a FLUX transformer and/or T5 encoder."""
"""Apply a LoRA model to a FLUX transformer and/or text encoder."""

lora: ModelIdentifierField = InputField(
description=FieldDescriptions.lora_model, title="LoRA", ui_type=UIType.LoRAModel
Expand Down
2 changes: 1 addition & 1 deletion invokeai/app/invocations/flux_text_encoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def _clip_encode(self, context: InvocationContext) -> torch.Tensor:
clip_text_encoder_config = clip_text_encoder_info.config
assert clip_text_encoder_config is not None

# Apply LoRA models to the T5 encoder.
# Apply LoRA models to the CLIP encoder.
# Note: We apply the LoRA after the transformer has been moved to its target device for faster patching.
if clip_text_encoder_config.format in [ModelFormat.Diffusers]:
# The model is non-quantized, so we can apply the LoRA weights directly into the model.
Expand Down
2 changes: 1 addition & 1 deletion invokeai/frontend/web/src/services/api/schema.ts
Original file line number Diff line number Diff line change
Expand Up @@ -6397,7 +6397,7 @@ export type components = {
};
/**
* FLUX LoRA
* @description Apply a LoRA model to a FLUX transformer and/or T5 encoder.
* @description Apply a LoRA model to a FLUX transformer and/or text encoder.
*/
FluxLoRALoaderInvocation: {
/**
Expand Down

0 comments on commit 536ccf0

Please # to comment.