Update README.md

README.md

@@ -9,8 +9,8 @@ tags:
 
 # DNA and Block Diffusion
 
-Using the [Block Diffusion](https://github.com/kuleshov-group/bd3lms) architecture and
-[AgroNT](https://huggingface.co/InstaDeepAI/agro-nucleotide-transformer-1b)'s six-nucleotide-length tokens.
+Using the **Block Diffusion** architecture ([code](https://github.com/kuleshov-group/bd3lms), [paper](https://arxiv.org/abs/2503.09573)) and
+[AgroNT](https://huggingface.co/InstaDeepAI/agro-nucleotide-transformer-1b)'s 6-nucleotide tokens (4^6 = 4,096).
 
 Took [dna-blockdiff-2](https://huggingface.co/monsoon-nlp/dna-blockdiff-2) weights,
 trained on [Papaya genome](https://huggingface.co/datasets/monsoon-nlp/wheat-bees) for one epoch. Output tokens are restricted to avoid outputting single-nucleotide or N (unknown nucleotide) tokens.