Update README.md

Update gemma-it prompt and format

README.md

@@ -3,37 +3,40 @@
 ---
 # Finetune Google Gemma model for Chinese Poem writing
 
-## Before finetune using Gemma-it-7b
+## Important! Gemma-IT format
 
 ```
-Before training
-上联：春风得意花铺路
-下联：花香满鼻，春暖
-
-上联：美丽中国魅力北京
-下联：中国传统建筑风格
-
-上联：鱼书千里梦
-下联：笑口在笑。
-
-对联：日落晚霞临古寺
-下联：花香满院，迎风
+GEMMA_IT_FORMAT = """<start_of_turn>user
+{user_question}<end_of_turn>
+<start_of_turn>model
+"""
 ```
 
-## After finetune (undertrained, fater 20mins using T4 GPU)
+## Before finetune using Gemma-it-7b
 
 ```
-上联：春风得意花铺路
-下联：秋雨飘落叶落落
-
-上联：美丽中国魅力北京
-下联：和谐世界和谐中国
+Before training
+春风得意花铺路
+春风得意花铺路，春暖花开，美不胜
+美丽中国魅力北京
+**美丽中国魅力北京，令人惊叹的现代与传统
+鱼书千里梦
+鱼书千里梦，是一个关于爱慕和成长的小说
+日落晚霞临古寺
+夕暮暮色，日落缓缓西下，将天空渲染成 Varies
+```
 
-上联：鱼书千里梦
-下联：鸟传万里风
+## After finetune (undertrained, after 10mins using T4 GPU, undertrained)
 
-对联：日落晚霞临古寺
-下联：风吹柳绿映新村
+```
+春风得意花铺路
+秋雨流派诗作韵<eos>
+美丽中国魅力北京
+和谐社会和谐社会<eos>
+鱼书千里梦
+鸟歌百载歌<eos>
+日落晚霞临古寺
+月明清风送客船<eos>
 ```
 
 ## Code to trigger this Lora ckpt (to be added)