2107
Browse files- 2b/diffusion_pytorch_model.safetensors +1 -1
- samples/2b_192x384_0.jpg +2 -2
- samples/2b_256x384_0.jpg +2 -2
- samples/2b_320x384_0.jpg +2 -2
- samples/2b_384x192_0.jpg +2 -2
- samples/2b_384x256_0.jpg +2 -2
- samples/2b_384x320_0.jpg +2 -2
- samples/2b_384x384_0.jpg +2 -2
- samples/unet_192x384_0.jpg +0 -3
- samples/unet_256x384_0.jpg +0 -3
- samples/unet_320x384_0.jpg +0 -3
- samples/unet_320x576_0.jpg +0 -3
- samples/unet_384x192_0.jpg +0 -3
- samples/unet_384x256_0.jpg +0 -3
- samples/unet_384x320_0.jpg +0 -3
- samples/unet_384x384_0.jpg +0 -3
- samples/unet_384x576_0.jpg +0 -3
- samples/unet_448x576_0.jpg +0 -3
- samples/unet_512x576_0.jpg +0 -3
- samples/unet_576x320_0.jpg +0 -3
- samples/unet_576x384_0.jpg +0 -3
- samples/unet_576x448_0.jpg +0 -3
- samples/unet_576x512_0.jpg +0 -3
- samples/unet_576x576_0.jpg +0 -3
- train.py +5 -5
2b/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7993399544
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f5d687a5ffb3a861067456a64fdf99ab51eb2a7ff8e9c98a80a60666b5bd6653
|
| 3 |
size 7993399544
|
samples/2b_192x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/2b_256x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/2b_320x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/2b_384x192_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/2b_384x256_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/2b_384x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/2b_384x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_192x384_0.jpg
DELETED
Git LFS Details
|
samples/unet_256x384_0.jpg
DELETED
Git LFS Details
|
samples/unet_320x384_0.jpg
DELETED
Git LFS Details
|
samples/unet_320x576_0.jpg
DELETED
Git LFS Details
|
samples/unet_384x192_0.jpg
DELETED
Git LFS Details
|
samples/unet_384x256_0.jpg
DELETED
Git LFS Details
|
samples/unet_384x320_0.jpg
DELETED
Git LFS Details
|
samples/unet_384x384_0.jpg
DELETED
Git LFS Details
|
samples/unet_384x576_0.jpg
DELETED
Git LFS Details
|
samples/unet_448x576_0.jpg
DELETED
Git LFS Details
|
samples/unet_512x576_0.jpg
DELETED
Git LFS Details
|
samples/unet_576x320_0.jpg
DELETED
Git LFS Details
|
samples/unet_576x384_0.jpg
DELETED
Git LFS Details
|
samples/unet_576x448_0.jpg
DELETED
Git LFS Details
|
samples/unet_576x512_0.jpg
DELETED
Git LFS Details
|
samples/unet_576x576_0.jpg
DELETED
Git LFS Details
|
train.py
CHANGED
|
@@ -30,9 +30,9 @@ project = "2b"
|
|
| 30 |
batch_size = 32
|
| 31 |
base_learning_rate = 8e-5
|
| 32 |
min_learning_rate = 4e-5
|
| 33 |
-
num_epochs =
|
| 34 |
# samples/save per epoch
|
| 35 |
-
sample_interval_share =
|
| 36 |
use_wandb = True
|
| 37 |
save_model = True
|
| 38 |
use_decay = True
|
|
@@ -52,9 +52,9 @@ save_barrier = 1.03
|
|
| 52 |
dispersive_temperature=0.5
|
| 53 |
dispersive_weight= 0.05
|
| 54 |
percentile_clipping = 95 # 8bit optim
|
| 55 |
-
betta2 = 0.
|
| 56 |
-
eps = 1e-
|
| 57 |
-
clip_grad_norm =
|
| 58 |
steps_offset = 1 # Scheduler
|
| 59 |
limit = 0
|
| 60 |
checkpoints_folder = ""
|
|
|
|
| 30 |
batch_size = 32
|
| 31 |
base_learning_rate = 8e-5
|
| 32 |
min_learning_rate = 4e-5
|
| 33 |
+
num_epochs = 27
|
| 34 |
# samples/save per epoch
|
| 35 |
+
sample_interval_share = 5
|
| 36 |
use_wandb = True
|
| 37 |
save_model = True
|
| 38 |
use_decay = True
|
|
|
|
| 52 |
dispersive_temperature=0.5
|
| 53 |
dispersive_weight= 0.05
|
| 54 |
percentile_clipping = 95 # 8bit optim
|
| 55 |
+
betta2 = 0.97
|
| 56 |
+
eps = 1e-5
|
| 57 |
+
clip_grad_norm = .75
|
| 58 |
steps_offset = 1 # Scheduler
|
| 59 |
limit = 0
|
| 60 |
checkpoints_folder = ""
|