Update README.md

README.md

@@ -77,7 +77,7 @@ Use the code below to get started with the model.
 
 ## Training procedure
 
-We employ a supervised-finetuning technique (SFT) on Axolotl with the following hyperparameters.
+We employ a supervised-finetuning technique (SFT) on Llama-factory with the following hyperparameters.
 
 ### Training hyperparameters
 
@@ -118,11 +118,11 @@ AUPRC is the primary metric to evaluate the safety classification of our models.
 **Environment & Requirements**
 | Library | Version |
 | :--- | :--- |
-| `Transformers` | `4.55.0` |
-| `PyTorch` | `2.6.0` |
-| `deepspeed` | `0.17.2` |
-| `accelerate` | `1.10.0` |
-| `axolotl` | `0.12.1` |
+| `Transformers` | `4.57.1` |
+| `PyTorch` | `2.7.1` |
+| `deepspeed` | `0.15.4` |
+| `accelerate` | `1.7.0` |
+| `llamafactory` | `0.9.4.dev0` |
 
 ## More Information