Update README.md

README.md

@@ -19,7 +19,7 @@ This pipeline was distilled from **SG161222/Realistic_Vision_V4.0** on a Subset
 ![val_imgs_grid](./grid_small.png)
 
 
-This Pipeline is based upon [the paper](https://arxiv.org/pdf/2305.15798.pdf). Training Code can be found [here](https://github.com/segmind/BKSDM).
+This Pipeline is based upon [the paper](https://arxiv.org/pdf/2305.15798.pdf). Training Code can be found [here](https://github.com/segmind/distill-sd).
 
 ## Pipeline usage