1
0
Fork 0

fix(api): add out channels to upscaling config

This commit is contained in:
Sean Sube 2023-11-18 21:35:57 -06:00
parent c8dd85e798
commit e64e514e7f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 0 deletions

View File

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