Update README.md

README.md

@@ -16,6 +16,10 @@ pipeline_tag: text-generation
 
 # HPC-Coder-v2
 
+> [!WARNING]
+> This is a midway training checkpoint for HPC-Coder-v2-6.7b.
+> The final weights are at [hpcgroup/hpc-coder-v2-6.7b](https://huggingface.co/hpcgroup/hpc-coder-v2-6.7b).
+
 The HPC-Coder-v2-6.7b model is an HPC code LLM fine-tuned on an instruction dataset catered to common HPC topics such as parallelism, optimization, accelerator porting, etc.
 This version is a fine-tuning of the [Deepseek Coder 6.7b](https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-base) model. 
 It is fine-tuned on the [hpc-instruct](https://huggingface.co/datasets/hpcgroup/hpc-instruct), [oss-instruct](https://huggingface.co/datasets/ise-uiuc/Magicoder-OSS-Instruct-75K), and [evol-instruct](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) datasets.