fixes nans by fixing index math in script

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:582ddd6c8b0660c32dde198ad362149f96c99d680677403e8be99a711366dcda
+oid sha256:c523486d1a6068cf86da801e1a1bb99dcccb314879f85b3ff2caf9efaa147b31
 size 9942981696

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d1e37e19a2eb827c2a21263d289e8873350fac9990f866a8b32c650131305754
+oid sha256:1431e04c3fc27af7fa60a8f77a7b4c7eb023831fd77ba6bc925274fc522da2f9
 size 4540516344