Update README.md

README.md

@@ -83,7 +83,7 @@ For ReFT, the nodes in the last 8 layers were unfrozen with attention to allow t
 
 After 3 to 4 epochs, the model began to overfit regardless of the strategies employed. Increasing both batch size and the number of epochs resulted in higher final training and evaluation cross-entropy.
 
-Following an extensive grid search, supervised fine-tuning of [Llama 3.1-8B](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) with LoRA+ and the parameters mentioned below yielded the best training and evaluation cross-entropy.
+Following an extensive grid search, supervised fine-tuning of [Llama 3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) with LoRA+ and the parameters mentioned below yielded the best training and evaluation cross-entropy.
 
 #### Preprocessing [optional]
 
@@ -105,7 +105,7 @@ Following an extensive grid search, supervised fine-tuning of [Llama 3.1-8B](htt
 - Number of epochs: 4
 
 
-#### Speeds, Sizes, Times [optional]
+#### Speeds, Sizes, Times
 
 This model was trained on ~550 million parameters on a training that lasted a bit more than 30 minutes and went through 4 epochs. The GPU utilization was above 90% at all times during training.
 
@@ -120,9 +120,9 @@ The final evaluation cross-entropy ended around 0.4.
 #### Metrics
 
 Since the fine-tuned model is designed to explain, and if possible, summarize newly learned data, ROUGE and BERTScore metrics were measured on a sample of 50 manually crafted questions. The reference answers were constructed during the creation of the training and evaluation sets. 
-Given that GPT-4-turbo was already used in this context, I did not compare my model against it. Instead, I chose to compare it against the following models:
+Given that GPT-4-turbo was already used in this context for the reference questions generation, I did not compare my model against it. Instead, I chose to compare it against the following models:
 
-| Metric | quantum-research-bot-v1.0 | Meta-Llama-3.1-8B  | gemini-1.5-pro   |
+| Metric | quantum-research-bot-v1.0 | Meta-Llama-3.1-8B-Instruct  | gemini-1.5-pro   |
 |:------------------|:---------------------------|:--------------------|:------------------|
 | **BERTScore F1**     | 0.5821                    | 0.3305             |    0.4982        |
 | **ROUGE-1** | 0.6045       | 0.3152    |0.5029  |