Skip to content

Commit 381c0ec

Browse files
committed
reviewer feedback
1 parent 09a4acb commit 381c0ec

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

examples/dreambooth/train_dreambooth_lora_qwen_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def save_model_card(
140140
model_card = load_or_create_model_card(
141141
repo_id_or_path=repo_id,
142142
from_training=True,
143-
license="mit", # TODO: Which license is this gonna be?
143+
license="apache-2.0",
144144
base_model=base_model,
145145
prompt=instance_prompt,
146146
model_description=model_description,

src/diffusers/loaders/lora_pipeline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6706,7 +6706,7 @@ def load_lora_weights(
67066706
)
67076707

67086708
@classmethod
6709-
# Copied from diffusers.loaders.lora_pipeline.SD3LoraLoaderMixin.load_lora_into_transformer with SD3Transformer2DModel->SanaTransformer2DModel
6709+
# Copied from diffusers.loaders.lora_pipeline.SD3LoraLoaderMixin.load_lora_into_transformer with SD3Transformer2DModel->QwenImageTransformer2DModel
67106710
def load_lora_into_transformer(
67116711
cls,
67126712
state_dict,
@@ -6725,7 +6725,7 @@ def load_lora_into_transformer(
67256725
A standard state dict containing the lora layer parameters. The keys can either be indexed directly
67266726
into the unet or prefixed with an additional `unet` which can be used to distinguish between text
67276727
encoder lora layers.
6728-
transformer (`SanaTransformer2DModel`):
6728+
transformer (`QwenImageTransformer2DModel`):
67296729
The Transformer model to load the LoRA layers into.
67306730
adapter_name (`str`, *optional*):
67316731
Adapter name to be used for referencing the loaded adapter model. If not specified, it will use

0 commit comments

Comments
 (0)