mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
Run ruff
This commit is contained in:
parent
42d606f07c
commit
a3dfa161a8
@ -1,6 +1,7 @@
|
||||
"""Model installation class."""
|
||||
|
||||
import os
|
||||
import pathlib
|
||||
import re
|
||||
import threading
|
||||
import time
|
||||
@ -10,9 +11,9 @@ from queue import Empty, Queue
|
||||
from shutil import copyfile, copytree, move, rmtree
|
||||
from tempfile import mkdtemp
|
||||
from typing import Any, Dict, List, Optional, Set, Union
|
||||
from omegaconf import DictConfig, OmegaConf
|
||||
|
||||
from huggingface_hub import HfFolder
|
||||
from omegaconf import DictConfig, OmegaConf
|
||||
from pydantic.networks import AnyHttpUrl
|
||||
from requests import Session
|
||||
|
||||
@ -297,7 +298,6 @@ class ModelInstallService(ModelInstallServiceBase):
|
||||
except OSError:
|
||||
return
|
||||
if len(db_models) == 0 and len(yaml.items()) != 0:
|
||||
|
||||
self._logger.info("No models in DB, yaml items need to be migrated")
|
||||
|
||||
for model_key, stanza in yaml.items():
|
||||
@ -308,11 +308,10 @@ class ModelInstallService(ModelInstallServiceBase):
|
||||
continue
|
||||
|
||||
_, _, model_name = str(model_key).split("/")
|
||||
import pathlib
|
||||
model_path = pathlib.Path(stanza["path"])
|
||||
if not model_path.is_absolute():
|
||||
model_path = (self._app_config.models_path / model_path)
|
||||
model_path.resolve()
|
||||
model_path = self._app_config.models_path / model_path
|
||||
model_path = model_path.resolve()
|
||||
description = stanza["description"]
|
||||
model_info = {"name": model_name, "description": description}
|
||||
|
||||
@ -321,7 +320,6 @@ class ModelInstallService(ModelInstallServiceBase):
|
||||
except Exception as e:
|
||||
self._logger.warning(f"Model at {model_path} could not be loaded into database: {e}")
|
||||
|
||||
|
||||
def scan_directory(self, scan_dir: Path, install: bool = False) -> List[str]: # noqa D102
|
||||
self._cached_model_paths = {Path(x.path).absolute() for x in self.record_store.all_models()}
|
||||
callback = self._scan_install if install else self._scan_register
|
||||
|
Loading…
Reference in New Issue
Block a user