Fix UnetParams annotation typo (#3589)

This commit is contained in:
JettHu 2024-05-28 07:30:35 +08:00 committed by GitHub
parent 0920e0e5fe
commit b26da2245f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -25,7 +25,7 @@ class UnetParams(TypedDict):
timestep: torch.Tensor
c: UnetApplyConds
# List of [0, 1], [0], [1], ...
# 0 means unconditional, 1 means conditional
# 0 means conditional, 1 means conditional unconditional
cond_or_uncond: List[int]