Nodes/ComfyUI-TITrain/Textual Inversion Training SD1.5
ComfyUI Node

Textual Inversion Training SD1.5

A ComfyUI node in train with 17 inputs and 0 outputs.

By mbrostami·Created 2 years ago·Updated about a year ago· 13
Textual Inversion Training SD1.5
  • images
    train_dirtrain
    pretrained_model_namerunwayml/stable-diffusion-v1-5
    learnable_propertyobject
    placeholder_tokencat-toy
    embedding_namecat-toy
    initializer_tokentoy
    resolution512
    max_train_steps500
    learning_rate0.0005
    mixed_precisionno
    validation_steps0
    num_vectors1
    batch_size1
    gradient_accumulation_steps4
    seed0
    prompt_extra
    Categorytrain

    Inputs (17)

    NameTypeDefaultDescription
    imagesIMAGE
    train_dirSTRINGtrain
    pretrained_model_nameSTRINGrunwayml/stable-diffusion-v1-5
    learnable_propertySTRINGobject
    placeholder_tokenSTRINGcat-toy
    embedding_nameSTRINGcat-toy
    initializer_tokenSTRINGtoy
    resolutionSTRING512
    max_train_stepsINT5001–8096
    learning_rateFLOAT0.0005
    mixed_precisionCOMBOno3 options: no, fp16, bf16
    validation_stepsINT00–2048
    num_vectorsINT11–15
    batch_sizeINT1
    gradient_accumulation_stepsINT4
    seedINT00–18446744073709550000
    prompt_extraSTRING

    Outputs (0)

    No outputs