Skip to content

Commit ff0b8e8

Browse files
Update README.md
1 parent ce52ea1 commit ff0b8e8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ For training conditional models we need two changes:
201201

202202
Specifically the dataset `getitem` will return the following:
203203
* `image_tensor` for unconditional training
204-
* tuple of `(image_tensor, cond_input )` for class conditional training where cond_input is a dictionary consisting of keys ```{class/text/image}```
204+
* tuple of `(image_tensor, cond_input )` for conditional training where cond_input is a dictionary consisting of keys ```{class/text/image}```
205205

206206
### Training Class Conditional LDM
207207
The repo provides class conditional latent diffusion model training code for mnist dataset, so one

0 commit comments

Comments
 (0)