Fix UnetParams annotation typo (#3589)

main
JettHu 9 months ago committed by GitHub
parent 0920e0e5fe
commit b26da2245f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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]

Loading…
Cancel
Save