Skip to content

Commit 7bd50ca

Browse files
authored
Add colab links to textual inversion (huggingface#375)
1 parent 5c4ea00 commit 7bd50ca

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

examples/textual_inversion/README.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,15 @@
33
[Textual inversion](https://arxiv.org/abs/2208.01618) is a method to personalize text2image models like stable diffusion on your own images using just 3-5 examples.
44
The `textual_inversion.py` script shows how to implement the training procedure and adapt it for stable diffusion.
55

6+
## Running on Colab
7+
8+
Colab for training
9+
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/sd_textual_inversion_training.ipynb)
10+
11+
Colab for inference
12+
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/stable_conceptualizer_inference.ipynb)
13+
14+
## Running locally
615
### Installing the dependencies
716

817
Before running the scipts, make sure to install the library's training dependencies:
@@ -64,7 +73,6 @@ A full training run takes ~1 hour on one V100 GPU.
6473

6574
Once you have trained a model using above command, the inference can be done simply using the `StableDiffusionPipeline`. Make sure to include the `placeholder_token` in your prompt.
6675

67-
6876
```python
6977

7078
from torch import autocast

0 commit comments

Comments
 (0)