Skip to content

Commit b767656

Browse files
committed
remove upscaler
1 parent 8065ea6 commit b767656

2 files changed

Lines changed: 0 additions & 55 deletions

File tree

finegrain/src/finegrain/__init__.py

Lines changed: 0 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -844,14 +844,6 @@ class BlendResultWithImage(OKResultWithImage, BlendResult):
844844
pass
845845

846846

847-
class UpscaleResult(OKResultWithUsedSeeds):
848-
pass
849-
850-
851-
class UpscaleResultWithImage(OKResultWithImage, UpscaleResult):
852-
pass
853-
854-
855847
class ShadowResult(OKResultWithUsedSeeds):
856848
@property
857849
def input_bbox(self) -> BoundingBox | None:
@@ -1151,34 +1143,6 @@ async def blend(
11511143
return await self._response_with_image(st, ok, BlendResultWithImage, params=image_params)
11521144
return await self._response(st, ok, BlendResult)
11531145

1154-
async def upscale(
1155-
self,
1156-
state_id: StateID,
1157-
preprocess: bool = True,
1158-
scale_factor: Literal[1, 2, 4] = 2,
1159-
resemblance: float | None = None,
1160-
decay: float | None = None,
1161-
creativity: float | None = None,
1162-
*,
1163-
seed: int | None = None,
1164-
with_image: bool | ImageOutParams = False,
1165-
timeout: float | None = None,
1166-
) -> UpscaleResult | ErrorResult:
1167-
params: dict[str, Any] = {"preprocess": preprocess, "scale_factor": scale_factor}
1168-
if resemblance is not None:
1169-
params["resemblance"] = resemblance
1170-
if decay is not None:
1171-
params["decay"] = decay
1172-
if creativity is not None:
1173-
params["creativity"] = creativity
1174-
if seed is not None:
1175-
params["seed"] = seed
1176-
st, ok = await self.ctx.call_skill(f"upscale/{state_id}", params, timeout=timeout)
1177-
if with_image:
1178-
image_params = None if isinstance(with_image, bool) else with_image
1179-
return await self._response_with_image(st, ok, UpscaleResultWithImage, params=image_params)
1180-
return await self._response(st, ok, UpscaleResult)
1181-
11821146
async def shadow(
11831147
self,
11841148
state_id: StateID,

finegrain/tests/test_upscale.py

Lines changed: 0 additions & 19 deletions
This file was deleted.

0 commit comments

Comments
 (0)