fix(api): add out channels to upscaling config
This commit is contained in:
parent
c8dd85e798
commit
e64e514e7f
|
@ -51,9 +51,15 @@ def preprocess(image):
|
|||
|
||||
|
||||
class FakeConfig:
|
||||
block_out_channels: List[int]
|
||||
scaling_factor: float
|
||||
|
||||
def __init__(self) -> None:
|
||||
self.block_out_channels = [
|
||||
128,
|
||||
256,
|
||||
512
|
||||
]
|
||||
self.scaling_factor = 0.08333
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue