Commit
•
852393f
1
Parent(s):
7d6e2bd
Fix beta schedule beta with the one used later on in training
Browse files
README.md
CHANGED
|
@@ -161,7 +161,7 @@ One problem with diffusion models is that they are [multi-task](https://arxiv.or
|
|
| 161 |
|
| 162 |
```py
|
| 163 |
alpha = 2.0
|
| 164 |
-
beta =
|
| 165 |
num_processes = self.accelerator.num_processes
|
| 166 |
process_index = self.accelerator.process_index
|
| 167 |
total_bsz = num_processes * bsz
|
|
|
|
| 161 |
|
| 162 |
```py
|
| 163 |
alpha = 2.0
|
| 164 |
+
beta = 1.6
|
| 165 |
num_processes = self.accelerator.num_processes
|
| 166 |
process_index = self.accelerator.process_index
|
| 167 |
total_bsz = num_processes * bsz
|