diff --git a/.gitignore b/.gitignore
index 8fe249c..afb89a2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,4 @@
 /__pycache__/
 /data/
 /music.safetensors
+/data.npz
diff --git a/dataInit.py b/dataInit.py
new file mode 100644
index 0000000..6622179
--- /dev/null
+++ b/dataInit.py
@@ -0,0 +1,13 @@
+import data
+import numpy as np
+
+x,y = data.dataset(data.load())
+size=len(x)
+
+x_np = np.stack(x)
+x_np = np.expand_dims(x_np, axis=1)
+
+y_np = np.stack(y)
+y_np = np.expand_dims(y_np, axis=1)
+
+np.savez_compressed("data",x_np,y_np)