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:
|
class FakeConfig:
|
||||||
|
block_out_channels: List[int]
|
||||||
scaling_factor: float
|
scaling_factor: float
|
||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
|
self.block_out_channels = [
|
||||||
|
128,
|
||||||
|
256,
|
||||||
|
512
|
||||||
|
]
|
||||||
self.scaling_factor = 0.08333
|
self.scaling_factor = 0.08333
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue