xSempiternus commited on
Commit
3ab1262
·
verified ·
1 Parent(s): 174aea6

Update README.md

Browse files

Training information added.

Files changed (1) hide show
  1. README.md +25 -1
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. -->