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
|