47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
from .nodes import (
|
|
AnimaBatchFolderLoader,
|
|
AnimaLlamaCppCaptionPrompt,
|
|
AnimaLoadCaptionImageBatch,
|
|
AnimaSaveLlamaCppCaptions,
|
|
AnimaVlmCaptionConfig,
|
|
AnimaBatchVlmCaption,
|
|
AnimaCaptionPrepare,
|
|
AnimaBatchTrainConfig,
|
|
AnimaBatchStartTrain,
|
|
AnimaBatchCheckStatus,
|
|
AnimaFindTrainedLora,
|
|
)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AnimaBatchFolderLoader": AnimaBatchFolderLoader,
|
|
"AnimaLlamaCppCaptionPrompt": AnimaLlamaCppCaptionPrompt,
|
|
"AnimaLoadCaptionImageBatch": AnimaLoadCaptionImageBatch,
|
|
"AnimaSaveLlamaCppCaptions": AnimaSaveLlamaCppCaptions,
|
|
"AnimaVlmCaptionConfig": AnimaVlmCaptionConfig,
|
|
"AnimaBatchVlmCaption": AnimaBatchVlmCaption,
|
|
"AnimaCaptionPrepare": AnimaCaptionPrepare,
|
|
"AnimaBatchTrainConfig": AnimaBatchTrainConfig,
|
|
"AnimaBatchStartTrain": AnimaBatchStartTrain,
|
|
"AnimaBatchCheckStatus": AnimaBatchCheckStatus,
|
|
"AnimaFindTrainedLora": AnimaFindTrainedLora,
|
|
}
|
|
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AnimaBatchFolderLoader": "Anima Batch Folder Loader",
|
|
"AnimaLlamaCppCaptionPrompt": "Anima Llama-cpp Caption Prompt",
|
|
"AnimaLoadCaptionImageBatch": "Anima Load Caption Image Batch",
|
|
"AnimaSaveLlamaCppCaptions": "Anima Save Llama-cpp Captions",
|
|
"AnimaVlmCaptionConfig": "Anima VLM Caption Config",
|
|
"AnimaBatchVlmCaption": "Anima Batch VLM Caption",
|
|
"AnimaCaptionPrepare": "Anima Caption Prepare",
|
|
"AnimaBatchTrainConfig": "Anima Batch Train Config",
|
|
"AnimaBatchStartTrain": "Anima Batch Start Train",
|
|
"AnimaBatchCheckStatus": "Anima Batch Check Status",
|
|
"AnimaFindTrainedLora": "Anima Find Trained LoRA",
|
|
}
|
|
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|