Update README.md

README.md

@@ -10,6 +10,10 @@ tags:
 - dct
 - swiglu
 - custom_code
+- base
+datasets:
+- wikimedia/wikipedia
+- HuggingFaceFW/fineweb-edu
 ---
 
 # KenT-300M-Base
@@ -41,6 +45,8 @@ D là ma trận DCT-II trực chuẩn cố định (hàng = tần số, vector h
 | tham số | ≈ 302.42M (embedding 67.55M) |
 | trạng thái giải mã | H shape (13, 1344), O(1) theo độ dài |
 
+![KenT diagram](KenT.png)
+
 EMA khởi tạo theo cấp số nhân từ 0.5 (tap đầu, feature đầu) đến 0.01 (tap cuối, feature cuối); vẫn học được.
 EMA chạy FP32 bằng affine-scan log-depth, kết quả bằng đúng hồi quy tuần tự.
 
@@ -50,4 +56,4 @@ tok = AutoTokenizer.from_pretrained("Vi-LAB/KenT-300M-Base", trust_remote_code=T
 model = AutoModelForCausalLM.from_pretrained("Vi-LAB/KenT-300M-Base", trust_remote_code=True)
 ```
 
-Hoặc `python infer.py --model Vi-LAB/KenT-300M-Base --prompt "Today the weather is"`.
+Hoặc `python infer.py --model Vi-LAB/KenT-300M-Base --prompt "Today the weather is"`.