Fix model card of LoRA (huggingface#2114)

Fix

Author: hysts

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:

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: