Upload . with huggingface_hub
Browse files- args.json +4 -4
- text_encoder/pytorch_model.bin +1 -1
- unet/diffusion_pytorch_model.bin +1 -1
args.json
CHANGED
|
@@ -12,15 +12,15 @@
|
|
| 12 |
"save_infer_steps": 50,
|
| 13 |
"pad_tokens": false,
|
| 14 |
"prior_loss_weight": 1.0,
|
| 15 |
-
"num_class_images":
|
| 16 |
"output_dir": "results",
|
| 17 |
"seed": 1234,
|
| 18 |
"resolution": 512,
|
| 19 |
"center_crop": false,
|
| 20 |
"train_batch_size": 1,
|
| 21 |
"sample_batch_size": 5,
|
| 22 |
-
"max_train_steps":
|
| 23 |
-
"max_train_text_encoder":
|
| 24 |
"gradient_checkpointing": false,
|
| 25 |
"learning_rate": 1e-06,
|
| 26 |
"lr_scheduler": "constant",
|
|
@@ -44,5 +44,5 @@
|
|
| 44 |
],
|
| 45 |
"repo_name": "mush",
|
| 46 |
"instance_class": "doy pack",
|
| 47 |
-
"num_train_epochs":
|
| 48 |
}
|
|
|
|
| 12 |
"save_infer_steps": 50,
|
| 13 |
"pad_tokens": false,
|
| 14 |
"prior_loss_weight": 1.0,
|
| 15 |
+
"num_class_images": 300,
|
| 16 |
"output_dir": "results",
|
| 17 |
"seed": 1234,
|
| 18 |
"resolution": 512,
|
| 19 |
"center_crop": false,
|
| 20 |
"train_batch_size": 1,
|
| 21 |
"sample_batch_size": 5,
|
| 22 |
+
"max_train_steps": 3000,
|
| 23 |
+
"max_train_text_encoder": 1500,
|
| 24 |
"gradient_checkpointing": false,
|
| 25 |
"learning_rate": 1e-06,
|
| 26 |
"lr_scheduler": "constant",
|
|
|
|
| 44 |
],
|
| 45 |
"repo_name": "mush",
|
| 46 |
"instance_class": "doy pack",
|
| 47 |
+
"num_train_epochs": 10
|
| 48 |
}
|
text_encoder/pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 680901463
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9be5dccd8d284080e6f8510d801b135e82e6815488e16292bf1fb87cbb5df901
|
| 3 |
size 680901463
|
unet/diffusion_pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1732101605
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:60cda4953f9e71a0858c175fdc576246438d1bf68c6e7bf12e4ba14ab15fedfd
|
| 3 |
size 1732101605
|