25ep
Browse files- result_grid.jpg +2 -2
- samples/unet_320x576_0.jpg +2 -2
- samples/unet_384x576_0.jpg +2 -2
- samples/unet_448x576_0.jpg +2 -2
- samples/unet_512x576_0.jpg +2 -2
- samples/unet_576x320_0.jpg +2 -2
- samples/unet_576x384_0.jpg +2 -2
- samples/unet_576x448_0.jpg +2 -2
- samples/unet_576x512_0.jpg +2 -2
- samples/unet_576x576_0.jpg +2 -2
- src/cherrypick.ipynb +1 -1
- test.ipynb +2 -2
- train.py +5 -1
- unet/diffusion_pytorch_model.fp16.safetensors +1 -1
- unet/diffusion_pytorch_model.safetensors +1 -1
result_grid.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_320x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_448x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_512x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x448_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x512_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
src/cherrypick.ipynb
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 44433
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c89da7c6029448005ad4789146ca5186d87824f9090a1e4310e87d9c014c9fcf
|
| 3 |
size 44433
|
test.ipynb
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:78d36025da61c85c0a801ce53028b6bea502ba64253a3eef66ab9827e64eb057
|
| 3 |
+
size 4784886
|
train.py
CHANGED
|
@@ -978,7 +978,11 @@ if accelerator.is_main_process:
|
|
| 978 |
# Сохраняем основную модель
|
| 979 |
if save_model:
|
| 980 |
save_checkpoint(unet,"fp16")
|
| 981 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 982 |
|
| 983 |
# randomize ode timesteps
|
| 984 |
# input_timestep = torch.round(
|
|
|
|
| 978 |
# Сохраняем основную модель
|
| 979 |
if save_model:
|
| 980 |
save_checkpoint(unet,"fp16")
|
| 981 |
+
accelerator.free_memory()
|
| 982 |
+
if torch.distributed.is_initialized():
|
| 983 |
+
torch.distributed.destroy_process_group()
|
| 984 |
+
|
| 985 |
+
print("Готово!")
|
| 986 |
|
| 987 |
# randomize ode timesteps
|
| 988 |
# input_timestep = torch.round(
|
unet/diffusion_pytorch_model.fp16.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3507231768
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:70a98b018dbffcb02789ed29bbe53823c888165df61a871f5c0eed1c951c774e
|
| 3 |
size 3507231768
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7014306128
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:90135cb9367dc964ea2e95ef59587c0a6baeb88a1e1baef10be027105160d20b
|
| 3 |
size 7014306128
|