Skip to content

Conversation

lstein
Copy link
Collaborator

@lstein lstein commented Jul 14, 2023

This PR adds several default models to the ones selected at install time. It also removes the GFPGAN and text2clip models, which should shave a little time off the install process.

ESRGAN:

  • models/core/upscaling/realesrgan/RealESRGAN_x4plus.pth
  • models/core/upscaling/realesrgan/RealESRGAN_x4plus_anime_6B.pth
  • models/core/upscaling/realesrgan/ESRGAN_SRx4_DF2KOST_official-ff704c30.pth

ControlNet

  • models/sd-1/controlnet/canny
  • models/sd-1/controlnet/depth
  • models/sd-1/controlnet/lineart
  • models/sd-1/controlnet/openpose

Embedding (textual inversion)

  • models/sd-1/embedding/EasyNegative.safetensors

- remove restoration services
- remove the restore faces nodes
- update tests
Copy link
Collaborator

@psychedelicious psychedelicious left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks Lincoln, I've added and tested a RealESRGAN upscaling node, works great.

We can fully remove the restoration services now, they are not needed. I've removed all references to them in the nodes app layer and updated the tests.

It's just the backend that has the actual restoration code - I'll leave that to you to remove.

@psychedelicious
Copy link
Collaborator

@maryhipp @brandonrising @ebr
This PR introduces preloading of:

It also introduces an upscaling node, which is not yet exposed in the Linear UI. I'll add that later.

Copy link
Contributor

@GreggHelt2 GreggHelt2 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@lstein
Copy link
Collaborator Author

lstein commented Jul 15, 2023

I've removed all the backend restoration code and will merge.

@lstein lstein enabled auto-merge July 15, 2023 22:27
@lstein lstein disabled auto-merge July 16, 2023 00:15
@psychedelicious
Copy link
Collaborator

I've added a couple small fixes:

  • Type the configuration service correctly as InvokeAIAppConfig
  • Update the realesrgan node to use the models_path var

@hipsterusername hipsterusername disabled auto-merge July 16, 2023 23:37
@hipsterusername hipsterusername merged commit 29afde7 into main Jul 16, 2023
@hipsterusername hipsterusername deleted the lstein/default-model-install branch July 16, 2023 23:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants