78 lines
2.1 KiB
Markdown
78 lines
2.1 KiB
Markdown
|
|
# Documentation
|
|
- Class name: Get Models
|
|
- Category: Bmad/dump
|
|
- Output node: True
|
|
- Repo Ref: https://github.com/comfyanonymous/ComfyUI
|
|
|
|
Get Models 节点旨在从指定目录中检索和组织各种模型相关文件。它支持多种模型类型和配置，使用户能够将全面的模型列表或特定类别的模型导出为 JSON 文件，以便于访问和管理。
|
|
|
|
# Input types
|
|
## Required
|
|
- dump
|
|
- 指定要检索的模型类别。可以是特定的模型类型，如"clip"或"vae"，也可以是"all models"以检索所有可用模型。这个选择决定了导出操作的范围，根据用户需求定制输出内容。
|
|
- Comfy dtype: COMBO[STRING]
|
|
- Python dtype: str
|
|
|
|
# Output types
|
|
该节点没有输出类型。
|
|
|
|
|
|
## Usage tips
|
|
- Infra type: `CPU`
|
|
- Common nodes: unknown
|
|
|
|
|
|
## Source code
|
|
```python
|
|
class GetModels:
|
|
dump_option = ['all models',
|
|
'checkpoints',
|
|
'clip',
|
|
'clip_vision',
|
|
'configs',
|
|
'controlnet',
|
|
'diffusers',
|
|
'embeddings',
|
|
'gligen',
|
|
'hypernetworks',
|
|
'loras',
|
|
'style_models',
|
|
'upscale_models',
|
|
'vae'
|
|
]
|
|
|
|
def __init__(self):
|
|
self.output_dir = folder_paths.get_output_directory()
|
|
self.type = "output"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {
|
|
"dump": (s.dump_option, {"default": "all models"})
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "dump_it"
|
|
CATEGORY = "Bmad/dump"
|
|
OUTPUT_NODE = True
|
|
|
|
def dump_it(self, dump):
|
|
dump_data = {}
|
|
|
|
if dump == 'all models':
|
|
for item in self.dump_option[1:]:
|
|
dump_data[item] = folder_paths.get_filename_list(item)
|
|
else:
|
|
dump_data['list'] = folder_paths.get_filename_list(dump)
|
|
|
|
file = f"{dump}.json"
|
|
file = os.path.join(self.output_dir, file)
|
|
with open(file, 'w') as f:
|
|
json.dump(dump_data, f, indent=1)
|
|
|
|
return ()
|
|
|
|
```
|