Fix UnetParams annotation typo (#3589)
This commit is contained in:
parent
0920e0e5fe
commit
b26da2245f
|
@ -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…
Reference in New Issue