install models/core/misc/lama/lama.pt

This commit is contained in:
Lincoln Stein 2023-09-04 21:54:40 -04:00
parent b1ef3370fa
commit d2e327add9

View File

@ -290,9 +290,19 @@ def download_realesrgan():
download_with_progress_bar(model["url"], config.models_path / model["dest"], model["description"])
# ---------------------------------------------
def download_lama():
logger.info("Installing lama infill model")
download_with_progress_bar(
'https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt',
config.models_path / 'core/misc/lama/lama.pt',
'lama infill model'
)
# ---------------------------------------------
def download_support_models():
download_realesrgan()
download_lama()
download_conversion_models()