recoilme commited on
Commit
8081318
·
1 Parent(s): 50e2534
result_grid.jpg CHANGED

Git LFS Details

  • SHA256: b4acaa192f5ea680276ed6f3fea739262fe15ddca72a415f9304e8a89c3feb7d
  • Pointer size: 132 Bytes
  • Size of remote file: 5.73 MB

Git LFS Details

  • SHA256: 36c6119732adbf585121db1dbefa51f6220813818fa0ff3733b0e018a35dabdf
  • Pointer size: 132 Bytes
  • Size of remote file: 5.6 MB
samples/unet_320x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 2f94dbd9a127d364431143b62867e72ee7ff00a38a75b538f716929a52c95d4d
  • Pointer size: 130 Bytes
  • Size of remote file: 62.2 kB

Git LFS Details

  • SHA256: de8007c35b54eb9645dfec2572ab0e31a6603da66363049d1e6587b797fb805c
  • Pointer size: 130 Bytes
  • Size of remote file: 55.1 kB
samples/unet_384x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 388b38ada560a3b9d0970a2fb3b3584a677fcce04f2f5888c42c048636daf0fc
  • Pointer size: 130 Bytes
  • Size of remote file: 68.3 kB

Git LFS Details

  • SHA256: ea0b4c23f5b1d7a2f77b15c33e81781d63ab4ab5644c00b499e52f7d98eeeab0
  • Pointer size: 130 Bytes
  • Size of remote file: 72 kB
samples/unet_448x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 9c28d7e457433acecbdae5a506dfabf8d33e5e8b14e22492f882cee915db5065
  • Pointer size: 131 Bytes
  • Size of remote file: 156 kB

Git LFS Details

  • SHA256: 667aaf68b9622b579db3ac7c5c2f1024219dd7df1e44c09a7a0cdd5f980ae649
  • Pointer size: 131 Bytes
  • Size of remote file: 134 kB
samples/unet_512x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 3c8012d5169b8d40200234975b4fe88c7e2b87e8cb7e8a1abfb3635aaa31026b
  • Pointer size: 130 Bytes
  • Size of remote file: 45.2 kB

Git LFS Details

  • SHA256: f04fa35f21b069a14797ae5371f3af1dd86617204fe1d42bb577a941e59b98bd
  • Pointer size: 130 Bytes
  • Size of remote file: 47.7 kB
samples/unet_576x320_0.jpg CHANGED

Git LFS Details

  • SHA256: 4f6814db508e97d689246531a81e11cc6378f5500dd690727b8c57ff9d00aa1a
  • Pointer size: 130 Bytes
  • Size of remote file: 73.5 kB

Git LFS Details

  • SHA256: 5d1d413850bb04608960bbe248f82ee71120dfeeff84129b7e543e5e9761a30d
  • Pointer size: 130 Bytes
  • Size of remote file: 61.9 kB
samples/unet_576x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 5f4a14010d79df5e64cea6c7b4f8aafa9af0aafd1cb4f51ef196ffb378a200b1
  • Pointer size: 130 Bytes
  • Size of remote file: 58.1 kB

Git LFS Details

  • SHA256: 65d9449551035a84868b046a043cbecb10f7a31f82b49b2b0ab312bbc4ddafd3
  • Pointer size: 130 Bytes
  • Size of remote file: 82.2 kB
samples/unet_576x448_0.jpg CHANGED

Git LFS Details

  • SHA256: 7e90eadc0dd44654587a1be286f278e4eae01ac2c00513a416f3c0e65492b07e
  • Pointer size: 131 Bytes
  • Size of remote file: 111 kB

Git LFS Details

  • SHA256: 90309cfa635cb883ef20871f61f9017800c42e264d6e69747464c5d28c594c37
  • Pointer size: 130 Bytes
  • Size of remote file: 98.5 kB
samples/unet_576x512_0.jpg CHANGED

Git LFS Details

  • SHA256: 72e5e5b8154e05faea75850317924fdf7e5d506205499f1171e45542dea71709
  • Pointer size: 131 Bytes
  • Size of remote file: 125 kB

Git LFS Details

  • SHA256: de2c60aae8d046bbff700ce627a229d6880392d61de47ce4bee84dbdbaf77818
  • Pointer size: 131 Bytes
  • Size of remote file: 103 kB
samples/unet_576x576_0.jpg CHANGED

Git LFS Details

  • SHA256: e5bcd730979e7dcab7c16d43352431b52d600eddd312957e6011ef6700a6d863
  • Pointer size: 131 Bytes
  • Size of remote file: 111 kB

Git LFS Details

  • SHA256: f6804ea3d511abb7069a07c6c70797f6a0d259e161356e21a1dbc0598cd934dc
  • Pointer size: 131 Bytes
  • Size of remote file: 119 kB
src/cherrypick.ipynb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0a7a1a0affd76326768c9aa37d032b8c0156a10f06f3fe53b98ca695a9dac57f
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:335c3f0398cd9cf6b3bda40a8f159c98b45e879087ba73502af4cb433fcc8373
3
- size 4966174
 
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
- print("Готово!")
 
 
 
 
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:19e1b82fe4bb97be41a71b12dc8b62d8310665f29ac44af690abe313431535d3
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:ea112ddf49eb35b24ad1e0be25a3867bdb68a2c8044ebd5954e14e395640ec36
3
  size 7014306128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90135cb9367dc964ea2e95ef59587c0a6baeb88a1e1baef10be027105160d20b
3
  size 7014306128