Update README.md

README.md

@@ -2,6 +2,9 @@
 
 See accompanying code at: https://github.com/davidbrandfonbrener/color-filter-olmo
 
+If you only want to download the filtered, untokenized data, see: https://huggingface.co/datasets/davidbrandfonbrener/color-filtered-c4
+
+## Usage
 
 To download the data, we recommend using the huggingface-cli.
 
@@ -18,10 +21,10 @@ If you only want to download some files (e.g. just the models), use the cli. For
 If you use this code in your research, please cite the following paper:
 
 ```bibtex
-@inproceedings{,
-  title={},
-  author={},
-  booktitle={},
-  year={},
+@article{brandfonbrener2024color,
+  title={CoLoR-Filter: Conditional Loss Reduction Filtering for Targeted Language Model Pre-training},
+  author={Brandfonbrener, David and Zhang, Hanlin and Kirsch, Andreas and Schwarz, Jonathan Richard and Kakade, Sham M},
+  journal={arXiv preprint arXiv:XXXX.XXXXX},
+  year={2024}
 }
 ```