Skip to content

Commit 0583101

Browse files
authored
Add Spandrel upscale starter models (#6605)
## Summary This PR adds some spandrel upscale models to the starter model list. In the future we may also want to add: - Some DAT models (https://drive.google.com/drive/folders/1iBdf_-LVZuz_PAbFtuxSKd_11RL1YKxM) ## QA Instructions I installed the starter models via the model manager UI, and tested that I could use them in a workflow. ## Merge Plan - [ ] Merge the preceding Spandrel PRs first, then change the target branch to `main`. ## Checklist - [x] _The PR has a short but descriptive title, suitable for a changelog_ - [x] _Tests added / updated (if applicable)_ - [x] _Documentation added / updated (if applicable)_
2 parents 95e9f53 + f866b49 commit 0583101

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)