Update README.md

README.md

@@ -13,7 +13,7 @@ This model is the instruction-tuned TESS 2. This model is a simplex-based diffus
 For more details, please check out our paper [TESS-2: A Large-Scale, Generalist Diffusion Language Model](https://todo).
 This is the model based on Mistral v0.1 and Tulu 2.
 
-This model will only work with our custom codebase found [here](https://github.com/armancohan/simplex-diffusion) -- please go there to see details on how to run training and inference.
+This model will only work with our custom codebase found [here](https://github.com/hamishivi/tess-2) -- please go there to see details on how to run training and inference.
 
 
 ## Using this model