diff --git a/src/model.py b/src/model.py
index 35ea089..494ce5a 100644
--- a/src/model.py
+++ b/src/model.py
@@ -1,3 +1,7 @@
+########################################################################################################
+# The RWKV Language Model - https://github.com/BlinkDL/RWKV-LM
+########################################################################################################
+
 import math
 import logging
 import numpy as np
diff --git a/train.py b/train.py
index 8374232..97ff024 100644
--- a/train.py
+++ b/train.py
@@ -1,3 +1,7 @@
+########################################################################################################
+# The RWKV Language Model - https://github.com/BlinkDL/RWKV-LM
+########################################################################################################
+
 import os, sys, time, math, random, json, datetime
 import logging
 import numpy as np