Update README.md

Ferg-Lab · Jan 20, 2023 · 7d09a25 · 7d09a25
1 parent 3a21793
commit 7d09a25
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -89,6 +89,8 @@ from molgen.models import DDPM
 model = DDPM(....)
 ```
 
+Code for the DDPM models are taken from: https://github.com/lucidrains/denoising-diffusion-pytorch (version 1.0.5)
+
 ### Copyright
 
 Copyright (c) 2023, Kirill Shmilovich