Fix beta schedule beta with the one used later on in training (#1)

- Fix beta schedule beta with the one used later on in training (852393fe4004ef17ffd8cabd7c45a9a503609a27)


Co-authored-by: Ryan Larimer <[email protected]>

README.md

@@ -161,7 +161,7 @@ One problem with diffusion models is that they are [multi-task](https://arxiv.or
 
 ```py
 alpha = 2.0
-beta = 2.0
+beta = 1.6
 num_processes = self.accelerator.num_processes
 process_index = self.accelerator.process_index
 total_bsz = num_processes * bsz