33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
"""
|
|
RandomLoRALoader Custom Node for ComfyUI
|
|
ランダムLoRA選択・適用ノード
|
|
"""
|
|
|
|
from .random_lora_loader import RandomLoRALoader
|
|
from .filtered_random_lora_loader import FilteredRandomLoRALoader
|
|
|
|
# LBW版は任意（読み込めない場合はスキップ）
|
|
try:
|
|
from .filtered_random_lora_loader_lbw import FilteredRandomLoRALoaderLBW
|
|
HAS_LBW = True
|
|
except Exception as e:
|
|
print(f"[RandomLoRALoader] Failed to load LBW version: {e}")
|
|
HAS_LBW = False
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RandomLoRALoader": RandomLoRALoader,
|
|
"FilteredRandomLoRALoader": FilteredRandomLoRALoader,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RandomLoRALoader": "Random LoRA Loader",
|
|
"FilteredRandomLoRALoader": "Filtered Random LoRA Loader",
|
|
}
|
|
|
|
# LBW版が利用可能な場合のみ追加
|
|
if HAS_LBW:
|
|
NODE_CLASS_MAPPINGS["FilteredRandomLoRALoaderLBW"] = FilteredRandomLoRALoaderLBW
|
|
NODE_DISPLAY_NAME_MAPPINGS["FilteredRandomLoRALoaderLBW"] = "Filtered Random LoRA Loader (LBW)"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|