Kratos Dreambooth

Created on January 16|Last edited on January 22
Comment
﻿
﻿
IntroductionThis an extremely spartan (pun intended) and minimalist WANDB report to document my progression with the attempt to incorporate Kratos 🪓 from God of War in Stable Diffusion using Dreambooth for the Hugging Face hackathon. 
﻿🖼️ Kratos dataset 
﻿🐍 Training notebook/script﻿
﻿🐝 Main WANDB run﻿
﻿🎨 Fine-tuned checkpoint (available on the Hub)
Config﻿
​
fearless-moon-17
meta(10 collapsed)
config
eval_batch_size
eval_batch_size
1
eval_samples_every_epochs
eval_samples_every_epochs
5
gradient_accumulation_steps
gradient_accumulation_steps
1
gradient_checkpointing
gradient_checkpointing
true
guidance_scale
guidance_scale
7
instance_prompt
instance_prompt
a photo of krts person
learning_rate
learning_rate
0.000002
max_grad_norm
max_grad_norm
1
max_train_steps
max_train_steps
880
output_dir
output_dir
my-dreambooth
pretrained_model_name_or_path
pretrained_model_name_or_path
CompVis/stable-diffusion-v1-4
project_name
project_name
dreambooth-kratos
resolution
resolution
512
sample_prompts
sample_prompts
["an illustration of krts person wearing a Santa hat in a snowy forest","a photograph of krts person reading a book on the beach","a sketch of krts person painting a mountain landscape","a drawing of krts person wearing a Spider-man costume in the style of Marvel comics"]
seed
seed
2077
train_batch_size
train_batch_size
1
train_dataset
train_dataset
<__main__.DreamBoothDataset at 0x7f063d04ca30>
train_text_encoder
train_text_encoder
true
summary
_wandb
Run set1
﻿
Summary metrics﻿
Run: fearless-moon-171
﻿
Sample outputs by epochTable showing the progression of the images generated, starting from generic illustrations to gradually produce images incorporating the new concept with more and more accuracy.
﻿
project("matt24", "dreambooth-kratos").artifact("run-ay51iqly-table").membershipForAlias("4263d6cbafa6276eff5f").artifactVersion.file("table.table.json")
 - 4 of 4
prompt
epoch_5
epoch_10
epoch_20
epoch_40
epoch_50
epoch_60
epoch_70
epoch_80
epoch_85
1
2
3
4
﻿
Final output samples
Punk﻿
project("matt24", "dreambooth-kratos").artifact("inference_artifact_2023-01-11_21-48-29").membershipForAlias("201fb2e98ca04d1e91e1").artifactVersion.file("inference_table.table.json")
 - 3 of 16
image
prompt
negative prompt
resolution
guidance scale
inference steps
seed
1
2
3
Spider-Man﻿
project("matt24", "dreambooth-kratos").artifact("inference_artifact_2023-01-11_21-38-22").membershipForAlias("v0").artifactVersion.file("inference_table.table.json")
 - 3 of 16
image
prompt
negative prompt
resolution
guidance scale
inference steps
seed
1
2
3
Movie theater﻿
project("matt24", "dreambooth-kratos").artifact("inference_artifact_2023-01-11_22-04-51").membershipForAlias("9a483b27cad816d499d2").artifactVersion.file("inference_table.table.json")
 - 3 of 16
image
prompt
negative prompt
resolution
guidance scale
inference steps
seed
1
2
3
Painting﻿
project("matt24", "dreambooth-kratos").artifact("inference_artifact_2023-01-11_21-32-10").membershipForAlias("3c15d6946ef14b262dd4").artifactVersion.file("inference_table.table.json")
 - 3 of 16
image
prompt
negative prompt
resolution
guidance scale
inference steps
seed
1
2
3
Futuristic﻿
project("matt24", "dreambooth-kratos").artifact("inference_artifact_2023-01-14_13-58-37").membershipForAlias("v0").artifactVersion.file("inference_table.table.json")
 - 3 of 16
image
prompt
negative prompt
resolution
guidance scale
inference steps
seed
1
2
3
﻿
Add a comment