Update README.md

README.md

@@ -19,6 +19,17 @@ Colab: [link](https://colab.research.google.com/drive/18GG-zwxyGAHoVXghlOQONkxb_
 
 Training code: [link](https://github.com/IlyaGusev/rulm/tree/master/self_instruct)
 
+Versions:
+- v2, April 8, 2023: A model trained with outputs capped at 800 tokens instead of 400. `revision="3dff86a20f80ba203fec6e986a3422b83822ca70"`
+  - Final loss: 0.995
+  - Time: 38 hours, 35 minutes
+  - Hardware: 1x RTX 3090Ti
+  - Wandb link: [link](https://wandb.ai/ilyagusev/rulm_self_instruct/runs/wy66bwim/overview?workspace=user-ilyagusev)
+- v1, April 2, 2023: Original model trained with 400 max tokens in outputs. Both Russian and English are supported, trained on the ru_turbo_alpaca and alpaca-cleaned. `revision="97209c36c9261eb9527371a341c70f90c39ee158"`
+  - Final loss: 1.074
+  - Time: 53 hours, 53 minutes
+  - Hardware: 1x RTX 3090
+
 ```python
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
@@ -50,8 +61,4 @@ Examples:
 "Здравствуйте, матушка", - сказала Таня. "Как вы знаете, где мой мяч?"
 "Я знаю", - ответила старая женщина. "Я видел, как он упал в воду, и я смог его вытащить. Я думаю, что он теперь готов к игре".
 Таня была счастлива, что ее любимой игрушеке удалось спастись
-```
-
-* Final loss: 1.074
-* Time: 53 hours, 53 minutes
-* Hardware: 1x RTX 3090
+```