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

add back the heuristic_import() method and extend repo_ids to arbitrary file paths

Browse Source
This commit is contained in:
Lincoln Stein
2024-02-11 23:37:49 -05:00
committed by psychedelicious
parent a23dedd2ee
commit 4027e845d4
6 changed files with 199 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/util/select_hf_files.py
View File

@ -36,6 +36,11 @@ def filter_files(
"""
variant = variant or ModelRepoVariant.DEFAULT
paths: List[Path] = []
root = files[0].parts[0]
# if the subfolder is a single file, then bypass the selection and just return it
if subfolder and subfolder.suffix in [".safetensors", ".bin", ".onnx", ".xml", ".pth", ".pt", ".ckpt", ".msgpack"]:
return [root / subfolder]
# Start by filtering on model file extensions, discarding images, docs, etc
for file in files:
@ -61,6 +66,7 @@ def filter_files(
# limit search to subfolder if requested
if subfolder:
subfolder = root / subfolder
paths = [x for x in paths if x.parent == Path(subfolder)]
# _filter_by_variant uniquifies the paths and returns a set