Added Inference Code

README.md

@@ -19,7 +19,7 @@ This pipeline was distilled from **SG161222/Realistic_Vision_V4.0** on a Subset
 ![val_imgs_grid](./grid_tiny.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
 
@@ -53,5 +53,7 @@ We have observed that the distilled models are upto 80% faster than the Base SD1
 ![graph](./graph.png)
 ![comparision](./comparision1.png)
 
-Below is the code for benchmarking the speeds
+[Here](https://github.com/segmind/distill-sd/blob/master/inference.py) is the code for benchmarking the speeds.
+
+