End of training

lr_scheduler/lr_scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ea8fcef1c10103b45bf245c16c4a32bfe64d418d3df5fc11ab5d9012c8613605
+oid sha256:042f23b825526b5511d45b2ade1e2ba8a5b7b4cec715ddadeef88b974a879767
 size 1076

optimizer/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:61eb4ceeb48038081812aa251522e1f89625f95574119563b3cad61a65655ec2
+oid sha256:84e060e24b854ae46cbafefb3d8f729bae383b2af5e7573c241d6d2725220539
 size 6876705882

scheduler/scheduler_config.py

@@ -31,7 +31,7 @@ class SDEParameterizedBaseLineConfig:
 
     initial_variable_value = 0
     max_variable_value = 1 # math.inf
-    min_sample_value = 1e-6
+    min_sample_value = 0
 
     module = 'jax'
 

scheduler/sdeparameters.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:78efaf25a7e28005ebaa25457db37513328376f9ab1129e585be27a9c81e6586
+oid sha256:603b6005bb3a30fae7ddc56681da6e2c909954cb4492a302fefb999cd9e70934
 size 140

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ba30a7475b6abdec9c65025b927e2505467535524e12723f7aee80b4b5623732
+oid sha256:a12b9d748210e482131db5a0c178932dc7a58dbca7201cc247c408375d2a6cc2
 size 3438144492