fffiloni commited on
Commit
4fd66b0
1 Parent(s): aba45a8

Update train_dreambooth_b-lora_sdxl.py

Browse files
Files changed (1) hide show
  1. train_dreambooth_b-lora_sdxl.py +6 -2
train_dreambooth_b-lora_sdxl.py CHANGED
@@ -142,9 +142,9 @@ license: openrail++
142
 
143
  ## Model description
144
 
145
- These are {repo_id} LoRA adaption weights for {base_model}.
146
 
147
- The weights were trained using [DreamBooth](https://dreambooth.github.io/).
148
 
149
  LoRA for the text encoder was enabled: {train_text_encoder}.
150
 
@@ -160,6 +160,10 @@ Weights for this model are available in Safetensors format.
160
 
161
  [Download]({repo_id}/tree/main) them in the Files & versions tab.
162
 
 
 
 
 
163
  """
164
  with open(os.path.join(repo_folder, "README.md"), "w") as f:
165
  f.write(yaml + model_card)
 
142
 
143
  ## Model description
144
 
145
+ These are {repo_id} B-LoRA adaption weights for {base_model}.
146
 
147
+ The weights were trained using [DreamBooth](https://dreambooth.github.io/).
148
 
149
  LoRA for the text encoder was enabled: {train_text_encoder}.
150
 
 
160
 
161
  [Download]({repo_id}/tree/main) them in the Files & versions tab.
162
 
163
+ ## Usage
164
+
165
+ If your model is public, you can use it with the [B-LoRa Inference space](https://huggingface.co/spaces/fffiloni/B-LoRa-Inference)
166
+
167
  """
168
  with open(os.path.join(repo_folder, "README.md"), "w") as f:
169
  f.write(yaml + model_card)