- Added upscale operator to handle image upscaling by 4x using AI models. - Introduced new jobs for upscale processing in the server, including debug and standard upscale jobs. - Created a new ONNX model handler for upscale operations, supporting tiled inference and resource management. - Updated tests to cover new upscale functionality, including input validation and output verification. - Added documentation for the upscale feature, detailing usage and model options.
15 lines
226 B
Python
15 lines
226 B
Python
from . import ben2, da3, moge2, upscale
|
|
|
|
|
|
def geometry_adapter(model_key):
|
|
return moge2 if model_key.startswith("MOGE2_") else da3
|
|
|
|
|
|
__all__ = (
|
|
"ben2",
|
|
"da3",
|
|
"moge2",
|
|
"upscale",
|
|
"geometry_adapter",
|
|
)
|