Skip to content

Commit d7619d4

Browse files
feat(mm): change anime upscaling model to one that doesn't trigger picklescan
1 parent 8ad4f6e commit d7619d4

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

invokeai/backend/model_manager/starter_models.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -580,13 +580,14 @@ class StarterModelBundle(BaseModel):
580580
)
581581
# endregion
582582
# region SpandrelImageToImage
583-
realesrgan_anime = StarterModel(
584-
name="RealESRGAN_x4plus_anime_6B",
583+
animesharp_v4_rcan = StarterModel(
584+
name="2x-AnimeSharpV4_RCAN",
585585
base=BaseModelType.Any,
586-
source="https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth",
587-
description="A Real-ESRGAN 4x upscaling model (optimized for anime images).",
586+
source="https://github.com/Kim2091/Kim2091-Models/releases/download/2x-AnimeSharpV4/2x-AnimeSharpV4_RCAN.safetensors",
587+
description="A 2x upscaling model (optimized for anime images).",
588588
type=ModelType.SpandrelImageToImage,
589589
)
590+
590591
realesrgan_x4 = StarterModel(
591592
name="RealESRGAN_x4plus",
592593
base=BaseModelType.Any,
@@ -732,7 +733,7 @@ class StarterModelBundle(BaseModel):
732733
t2i_lineart_sdxl,
733734
t2i_sketch_sdxl,
734735
realesrgan_x4,
735-
realesrgan_anime,
736+
animesharp_v4_rcan,
736737
realesrgan_x2,
737738
swinir,
738739
t5_base_encoder,

0 commit comments

Comments
 (0)