add AnyText

iopaint/download.py

@@ -12,6 +12,7 @@ from iopaint.const import (
     DIFFUSERS_SD_INPAINT_CLASS_NAME,
     DIFFUSERS_SDXL_CLASS_NAME,
     DIFFUSERS_SDXL_INPAINT_CLASS_NAME,
+    ANYTEXT_NAME,
 )
 from iopaint.model_info import ModelInfo, ModelType
 
@@ -24,6 +25,10 @@ def cli_download_model(model: str):
         logger.info(f"Downloading {model}...")
         models[model].download()
         logger.info(f"Done.")
+    elif model == ANYTEXT_NAME:
+        logger.info(f"Downloading {model}...")
+        models[model].download()
+        logger.info(f"Done.")
     else:
         logger.info(f"Downloading model from Huggingface: {model}")
         from diffusers import DiffusionPipeline
@@ -210,6 +215,7 @@ def scan_models() -> List[ModelInfo]:
                 "StableDiffusionInstructPix2PixPipeline",
                 "PaintByExamplePipeline",
                 "KandinskyV22InpaintPipeline",
+                "AnyText",
             ]:
                 model_type = ModelType.DIFFUSERS_OTHER
             else: