moizsajid/dreambooth-markhor
Viewer • Updated • 5 • 28
How to use keras-dreambooth/dreambooth-markhor with Keras:
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"
import keras
model = keras.saving.load_model("hf://keras-dreambooth/dreambooth-markhor")
This is a Stable Diffusion model fine-tuned using Dreambooth on markhor images.
Submission for keras-dreambooth sprint
More information needed
The following hyperparameters were used during training:
| Hyperparameters | Value |
|---|---|
| name | RMSprop |
| weight_decay | None |
| clipnorm | None |
| global_clipnorm | None |
| clipvalue | None |
| use_ema | False |
| ema_momentum | 0.99 |
| ema_overwrite_frequency | 100 |
| jit_compile | True |
| is_legacy_optimizer | False |
| learning_rate | 0.0010000000474974513 |
| rho | 0.9 |
| momentum | 0.0 |
| epsilon | 1e-07 |
| centered | False |
| training_precision | float32 |