nodes/Zuellni-ComfyUI-ExLlama-Nodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #14 from ScottNealon/patch-1

Browse Source 
Parse all possible folders mapped to "llm"
This commit is contained in:
Zuellni
2024-02-03 15:11:39 +01:00
committed by GitHub
parent 73e6a0b0e4 348aeea258
commit 261a73a119
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
exllama.py
+6 -6
View File
@@ -20,12 +20,12 @@ from folder_paths import folder_names_and_paths, get_folder_paths, models_dir
class Loader: class Loader:
@classmethod @classmethod
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
if not "llm" in folder_names_and_paths: add_model_folder_path("llm", str(Path(models_dir).joinpath("llm")))
folder_names_and_paths["llm"] = ([str(Path(models_dir) / "llm")],) for folder in get_folder_paths("llm"):
if Path(folder).is_dir():
for path in Path(get_folder_paths("llm")[0]).glob("*/"): for path in Path(folder).glob("*/"):
if (path / "config.json").is_file(): if (path / "config.json").is_file():
cls._MODELS[path.name] = path cls._MODELS[path.name] = path
models = list(cls._MODELS.keys()) models = list(cls._MODELS.keys())
default = models[0] if models else None default = models[0] if models else None