Fix model card of LoRA (huggingface#2114)

Fix
itadapter · Jan 26, 2023 · 7436e30 · 7436e30
1 parent 1497650
commit 7436e30
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py
@@ -81,7 +81,7 @@ def save_model_card(repo_name, images=None, base_model=str, prompt=str, repo_fol
     model_card = f"""
 # LoRA DreamBooth - {repo_name}
 
-These are LoRA adaption weights for {repo_name}. The weights were trained on {prompt} using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. \n
+These are LoRA adaption weights for {base_model}. The weights were trained on {prompt} using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. \n
 {img_str}
 """
     with open(os.path.join(repo_folder, "README.md"), "w") as f:

diff --git a/examples/text_to_image/train_text_to_image_lora.py b/examples/text_to_image/train_text_to_image_lora.py
@@ -73,7 +73,7 @@ def save_model_card(repo_name, images=None, base_model=str, dataset_name=str, re
     """
     model_card = f"""
 # LoRA text2image fine-tuning - {repo_name}
-These are LoRA adaption weights for {repo_name}. The weights were fine-tuned on the {dataset_name} dataset. You can find some example images in the following. \n
+These are LoRA adaption weights for {base_model}. The weights were fine-tuned on the {dataset_name} dataset. You can find some example images in the following. \n
 {img_str}
 """
     with open(os.path.join(repo_folder, "README.md"), "w") as f: