Linoy Tsaban
commited on
Commit
•
53c557d
1
Parent(s):
a6fc407
Update README.md
Browse files
README.md
CHANGED
|
@@ -14,6 +14,20 @@ This is a Stable Diffusion model fine-tuned on the groot concept with DreamBooth
|
|
| 14 |
|
| 15 |
This model was created as part of the DreamBooth Hackathon 🔥. Visit the [organisation page](https://huggingface.co/dreambooth-hackathon) for instructions on how to take part!
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
## Usage
|
| 18 |
|
| 19 |
```python
|
|
|
|
| 14 |
|
| 15 |
This model was created as part of the DreamBooth Hackathon 🔥. Visit the [organisation page](https://huggingface.co/dreambooth-hackathon) for instructions on how to take part!
|
| 16 |
|
| 17 |
+
params used:
|
| 18 |
+
scheduler- DDIMScheduler
|
| 19 |
+
resolution=512
|
| 20 |
+
learning_rate=2e-06
|
| 21 |
+
max_train_steps= 500
|
| 22 |
+
train_batch_size=1
|
| 23 |
+
gradient_accumulation_steps=2
|
| 24 |
+
max_grad_norm=1.0
|
| 25 |
+
gradient_checkpointing=True
|
| 26 |
+
use_8bit_adam=True
|
| 27 |
+
seed=14071995,
|
| 28 |
+
sample_batch_size=2
|
| 29 |
+
|
| 30 |
+
|
| 31 |
## Usage
|
| 32 |
|
| 33 |
```python
|