diff --git a/README.md b/README.md index d657a1c..3336f77 100644 --- a/README.md +++ b/README.md @@ -70,6 +70,8 @@ Applying to random noise produces ![Random Noise]() # Associative Dictionary +This approach currently does not work so I'll need to think about it for a while, but it +does run! This is an attempt to store more than one image in a network based on text embedding and associated image. In principle it could also be a generative model if you ask for something not in the dictionary, but we'll see what happens. I'm using the [Pick-a-Pic](https://stability.ai/research/pick-a-pic) so you'll need to download those parquet files - the idea here is not (yet) to train on an enormous dataset but maybe use 10s to 100s of images. ``` python3 examples/text_to_image.py batch_size=2000 optimizer=sparse_lion max_epochs=10