Modelpublic

keras-dreambooth/marvin_paranoid_android

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face

1likes10downloads

Model Card

Model description

This model is a fine-tuned Stable Diffusion modeled, using the Dreambooth technique. It was trained on 15 images of Marvin the Paranoid Android, scraped from the Internet, as instance images and 205 robot images generated by Stable Diffusion as class images. You can find the full set here: Marvin the Paranoid Android

It was created by johko for the keras-dreambooth sprint.

Training procedure

This model was trained using the keras implementation of dreambooth. You can find the notebook to train these models and how to attend this sprint here.

Example Outputs

[image] [image] [image]

Training hyperparameters

The following hyperparameters were used during training:

Hyperparameters	Value
inneroptimizer.classname	Custom>RMSprop
inner_optimizer.config.name	RMSprop
inneroptimizer.config.weightdecay	None
inner_optimizer.config.clipnorm	None
inneroptimizer.config.globalclipnorm	None
inner_optimizer.config.clipvalue	None
inneroptimizer.config.useema	False
inneroptimizer.config.emamomentum	0.99
inneroptimizer.config.emaoverwrite_frequency	100
inneroptimizer.config.jitcompile	True
inneroptimizer.config.islegacy_optimizer	False
inneroptimizer.config.learningrate	0.0010000000474974513
inner_optimizer.config.rho	0.9
inner_optimizer.config.momentum	0.0
inner_optimizer.config.epsilon	1e-07
inner_optimizer.config.centered	False
dynamic	True
initial_scale	32768.0
dynamicgrowthsteps	2000
training_precision	mixed_float16

## Model Plot

<details> <summary>View Model Plot</summary>

[image]

</details>