diff --git a/gpt2.py b/gpt2.py
index c85e8b8..ea40714 100644
--- a/gpt2.py
+++ b/gpt2.py
@@ -1,3 +1,4 @@
+## Import numpy error
 import numpy as np