Fix beta schedule beta with the one used later on in training

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