ComfyUI Node: Textual Inversion Training SD1.5

Authored by mbrostami

Created

Updated

8 stars

Category

train

Inputs

images IMAGE
train_dir STRING
pretrained_model_name STRING
learnable_property STRING
placeholder_token STRING
embedding_name STRING
initializer_token STRING
resolution STRING
max_train_steps INT
learning_rate FLOAT
mixed_precision
  • no
  • fp16
  • bf16
validation_steps INT
num_vectors INT
batch_size INT
gradient_accumulation_steps INT
seed INT
prompt_extra STRING

Outputs

Extension: ComfyUI-TITrain

Nodes:TextualInversionTrainingSDXL, TextualInversionTraining

Authored by mbrostami

Run ComfyUI workflows in the Cloud!

No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

Learn more