• allo-'s avatar
    [textual_inversion] Add an option for only saving the embeddings (#781) · d1bcbf38
    allo- authored
    [textual_inversion] Add an option to only save embeddings
    
    Add an command line option --only_save_embeds to the example script, for
    not saving the full model. Then only the learned embeddings are saved,
    which can be added to the original model at runtime in a similar way as
    they are created in the training script.
    Saving the full model is forced when --push_to_hub is used. (Implements #759)
    d1bcbf38
textual_inversion.py 24.3 KB