Update README.md

README.md

@@ -14,7 +14,7 @@ tags:
 
 These are the checkpoints for the first ever QLoRA for ESM-2! They haven't been checked for overfitting yet, so use with caution!
 You can load and use them similarly to the LoRA models. This is the smallest `esm2_t6_8M_UR50D` model, so the metrics aren't great. 
-Scaling to larger models for better metrics is in progress. 
+Scaling to larger models for better metrics is in progress. These checkpoints were trained using [the 600K dataset](https://huggingface.co/datasets/AmelieSchreiber/600K_data). 
 
 ## QLoRA Info