Unverified Commit b26da224 authored by JettHu's avatar JettHu Committed by GitHub
Browse files

Fix UnetParams annotation typo (#3589)

parent 0920e0e5
...@@ -25,7 +25,7 @@ class UnetParams(TypedDict): ...@@ -25,7 +25,7 @@ class UnetParams(TypedDict):
timestep: torch.Tensor timestep: torch.Tensor
c: UnetApplyConds c: UnetApplyConds
# List of [0, 1], [0], [1], ... # List of [0, 1], [0], [1], ...
# 0 means unconditional, 1 means conditional # 0 means conditional, 1 means conditional unconditional
cond_or_uncond: List[int] cond_or_uncond: List[int]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment