Update README.md
Browse filesTraining information added.
README.md
CHANGED
|
@@ -1,5 +1,12 @@
|
|
| 1 |
---
|
| 2 |
library_name: diffusers
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
# Model Card for Model ID
|
|
@@ -73,7 +80,24 @@ Use the code below to get started with the model.
|
|
| 73 |
[More Information Needed]
|
| 74 |
|
| 75 |
## Training Details
|
| 76 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 77 |
### Training Data
|
| 78 |
|
| 79 |
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|
|
|
|
| 1 |
---
|
| 2 |
library_name: diffusers
|
| 3 |
+
base_model:
|
| 4 |
+
- CompVis/stable-diffusion-v1-4
|
| 5 |
+
pipeline_tag: text-to-image
|
| 6 |
+
tags:
|
| 7 |
+
- cat
|
| 8 |
+
- text-to-image
|
| 9 |
+
- art
|
| 10 |
---
|
| 11 |
|
| 12 |
# Model Card for Model ID
|
|
|
|
| 80 |
[More Information Needed]
|
| 81 |
|
| 82 |
## Training Details
|
| 83 |
+
!accelerate launch diffusers/examples/dreambooth/train_dreambooth.py \
|
| 84 |
+
--pretrained_model_name_or_path="CompVis/stable-diffusion-v1-4" \
|
| 85 |
+
--instance_data_dir="/content/Dataset" \
|
| 86 |
+
--output_dir="/content/canelo-cat-finetunedv1" \
|
| 87 |
+
--instance_prompt="a photo of a sks cat" \
|
| 88 |
+
--resolution=512 \
|
| 89 |
+
--train_batch_size=1 \
|
| 90 |
+
--gradient_accumulation_steps=4 \
|
| 91 |
+
--learning_rate=2e-6 \
|
| 92 |
+
--lr_scheduler="constant" \
|
| 93 |
+
--mixed_precision="fp16" \
|
| 94 |
+
--use_8bit_adam \
|
| 95 |
+
--gradient_checkpointing \
|
| 96 |
+
--max_train_steps=150 \
|
| 97 |
+
--checkpointing_steps=50 \
|
| 98 |
+
--seed=42 \
|
| 99 |
+
--validation_prompt="a photo of a sks cat in a park" \
|
| 100 |
+
--validation_steps=50
|
| 101 |
### Training Data
|
| 102 |
|
| 103 |
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|