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

added hf models import tab and route for getting available hf models

Browse Source
This commit is contained in:
Jennifer Player
2024-03-07 16:57:28 -05:00
committed by psychedelicious
parent efea1a8a7d
commit f7cd3cf1f4
10 changed files with 308 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,7 +51,15 @@ def filter_files(
(
"learned_embeds.bin",
"ip_adapter.bin",
"lora_weights.safetensors",
# jennifer added a bunch of these, probably will break something
".safetensors",
".bin",
".onnx",
".xml",
".pth",
".pt",
".ckpt",
".msgpack",
"weights.pb",
"onnx_data",
)
@ -71,7 +79,8 @@ def filter_files(
paths = [x for x in paths if x.parent == Path(subfolder)]
# _filter_by_variant uniquifies the paths and returns a set
return sorted(_filter_by_variant(paths, variant))
# jennifer removed the filter since it removed models, probably will break something but i dont understand why it removes valid models :|
return sorted(paths)
@dataclass