End of training

optimizer/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:31f24ae8c9be356b80823017a6443f945de9d160bf797c00ddb5c588e5ddf595
+oid sha256:50f22f9265e94e528478207e4fc002476ad0034abeea03c61fa14773b8587fff
 size 1149290878

scheduler/scheduler_config.py

@@ -52,4 +52,4 @@ class SDEPolynomialConfig:
 
 
     target = "epsilon" # x0
-    non_symbolic_parameters = {'drift': torch.ones(drift_degree)}
+    non_symbolic_parameters = {'drift': torch.ones(drift_degree), 'diffusion': torch.tensor([1.])}

scheduler/sdeparameters.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b59e815a0d5d903799db1da0cfaa61298044b04691741e1b76dd1404f6e6d38
+oid sha256:eeb1459bbcbc570e1500ace6bd35966cdd2d7991ab5ad91635bdf3d0080d7acf
 size 220

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:77aac723d4efa78d5f3f90d242c24cc384f2826c2459f7e2ee1b3e3dc2f8f022
+oid sha256:639d03b452944d97c7f29345d8d18affc241952a3f70890cc923570fa28e5448
 size 574476604