@@ -399,6 +399,43 @@ class StarterModel(StarterModelWithoutDependencies):
399
399
type = ModelType .T2IAdapter ,
400
400
),
401
401
# 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
402
439
]
403
440
404
441
assert len (STARTER_MODELS ) == len ({m .source for m in STARTER_MODELS }), "Duplicate starter models"
0 commit comments