This commit is contained in:
Qing
2024-01-05 15:38:34 +08:00
parent a95dd10ceb
commit cc06f30e9a
6 changed files with 41 additions and 19 deletions

View File

@@ -15,11 +15,9 @@ from iopaint.const import (
)
from iopaint.model.utils import handle_from_pretrained_exceptions
from iopaint.model_info import ModelInfo, ModelType
from iopaint.runtime import setup_model_dir
def cli_download_model(model: str, model_dir: Path):
setup_model_dir(model_dir)
def cli_download_model(model: str):
from iopaint.model import models
if model in models and models[model].is_erase_model: