Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Run ruff

Browse Source
This commit is contained in:
brandonrising 2024-03-25 20:59:34 -04:00 committed by Brandon
parent 90f32d17ba
commit 653a8f67a4
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/probe.py
View File

@ -649,7 +649,11 @@ class VaeFolderProbe(FolderProbeBase):
class TextualInversionFolderProbe(TextualInversionCheckpointProbe):
def __init__(self, model_path: Path):
files = os.scandir(model_path)
files = [Path(f.path) for f in files if f.is_file() and f.name.endswith((".ckpt", ".pt", ".pth", ".bin", ".safetensors"))]
files = [
Path(f.path)
for f in files
if f.is_file() and f.name.endswith((".ckpt", ".pt", ".pth", ".bin", ".safetensors"))
]
if len(files) != 1:
raise InvalidModelConfigException(
f"Unable to determine base type for {model_path}: expected exactly one valid model file, found {[f.name for f in files]}."