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. -->
|