Unverified Commit 7436e30c authored by hysts's avatar hysts Committed by GitHub
Browse files

Fix model card of LoRA (#2114)

Fix
parent 14976500
......@@ -81,7 +81,7 @@ inference: true
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:
......
......@@ -73,7 +73,7 @@ inference: true
"""
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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment