54 lines
1.6 KiB
Markdown
54 lines
1.6 KiB
Markdown
|
|
# Documentation
|
|
- Class name: AV_VAELoader
|
|
- Category: Art Venture/Loaders
|
|
- Output node: False
|
|
|
|
AV_VAELoader节点旨在加载VAE模型，并提供可选的覆盖功能，允许用户指定替代的VAE模型进行加载。这一功能增强了Art Venture框架内模型选择和使用的灵活性。
|
|
|
|
# Input types
|
|
## Required
|
|
- vae_name
|
|
- 要加载的VAE模型名称。除非指定了覆盖模型，否则这将作为默认加载的模型。
|
|
- Comfy dtype: COMBO[STRING]
|
|
- Python dtype: str
|
|
## Optional
|
|
- vae_override
|
|
- 允许指定替代的VAE模型进行加载。如果找不到指定的模型，则会加载默认或预期的VAE模型，从而提供了灵活的模型选择方法。
|
|
- Comfy dtype: STRING
|
|
- Python dtype: str
|
|
|
|
# Output types
|
|
- vae
|
|
- 加载完成的VAE模型，可在系统中直接使用。这个输出使得能够进行进一步的操作或使用该模型进行转换。
|
|
- Comfy dtype: VAE
|
|
- Python dtype: torch.nn.Module
|
|
|
|
|
|
## Usage tips
|
|
- Infra type: `CPU`
|
|
- Common nodes: unknown
|
|
|
|
|
|
## Source code
|
|
```python
|
|
class AVVAELoader(VAELoader):
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
inputs = VAELoader.INPUT_TYPES()
|
|
inputs["optional"] = {"vae_override": ("STRING", {"default": "None"})}
|
|
return inputs
|
|
|
|
CATEGORY = "Art Venture/Loaders"
|
|
|
|
def load_vae(self, vae_name, vae_override="None"):
|
|
if vae_override != "None":
|
|
if vae_override not in folder_paths.get_filename_list("vae"):
|
|
print(f"Warning: Not found VAE model {vae_override}. Use {vae_name} instead.")
|
|
else:
|
|
vae_name = vae_override
|
|
|
|
return super().load_vae(vae_name)
|
|
|
|
```
|