Skip to content

Commit f866b49

Browse files
committed
Add some ESRGAN and SwinIR upscale models to the starter models list.
1 parent 95e9f53 commit f866b49

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed

invokeai/backend/model_manager/starter_models.py

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -399,6 +399,43 @@ class StarterModel(StarterModelWithoutDependencies):
399399
type=ModelType.T2IAdapter,
400400
),
401401
# endregion
402+
# region SpandrelImageToImage
403+
StarterModel(
404+
name="RealESRGAN_x4plus_anime_6B",
405+
base=BaseModelType.Any,
406+
source="https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth",
407+
description="A Real-ESRGAN 4x upscaling model (optimized for anime images).",
408+
type=ModelType.SpandrelImageToImage,
409+
),
410+
StarterModel(
411+
name="RealESRGAN_x4plus",
412+
base=BaseModelType.Any,
413+
source="https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth",
414+
description="A Real-ESRGAN 4x upscaling model (general-purpose).",
415+
type=ModelType.SpandrelImageToImage,
416+
),
417+
StarterModel(
418+
name="ESRGAN_SRx4_DF2KOST_official",
419+
base=BaseModelType.Any,
420+
source="https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/ESRGAN_SRx4_DF2KOST_official-ff704c30.pth",
421+
description="The official ESRGAN 4x upscaling model.",
422+
type=ModelType.SpandrelImageToImage,
423+
),
424+
StarterModel(
425+
name="RealESRGAN_x2plus",
426+
base=BaseModelType.Any,
427+
source="https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth",
428+
description="A Real-ESRGAN 2x upscaling model (general-purpose).",
429+
type=ModelType.SpandrelImageToImage,
430+
),
431+
StarterModel(
432+
name="SwinIR - realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN",
433+
base=BaseModelType.Any,
434+
source="https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN-with-dict-keys-params-and-params_ema.pth",
435+
description="A SwinIR 4x upscaling model.",
436+
type=ModelType.SpandrelImageToImage,
437+
),
438+
# endregion
402439
]
403440

404441
assert len(STARTER_MODELS) == len({m.source for m in STARTER_MODELS}), "Duplicate starter models"

0 commit comments

Comments
 (0)