utput_dir="textual_inversion_cat"
💡 If you want to increase the trainable capacity, you can associate your placeholder token, e.g. <cat-toy> to
multiple embedding vectors. This can help the model to better capture the style of more (complex) images.
To enable training multiple embedding vectors, simply pass:


	Copied
--num_vectors=5

JAX

Hide JAX content

If you have access to TPUs, try out the Flax training script to train even faster (this’