archae-ai / readme.md
sgraham's picture
Update readme.md
30292bc
|
raw
history blame
1.81 kB
Current best model from our experiments to finetune CLIP on 5k archaeological record photos plus another 300 from the Met.
See [blog post 1](https://carleton.ca/xlab/2023/archaeclip-or-building-a-visual-search-engine-for-archaeology/) and a companion post at [open context](https://alexandriaarchive.org/2023/10/08/artificial-intelligence-ai-and-open-context/)
The goal is to use with `LLM`, Simon Willison's package for working with large language models, and in particular, `LLM-CLIP`, to make our own embeddings-powered search engine.
https://github.com/simonw/llm-clip
requires LLM: https://llm.datasette.io/en/stable/
So:
```
$ pip install llm
$ llm install llm-clip
```
Then, assuming you are doing this in an environment (I create mine with conda), find the site packages directory, and the llm-clip.py file:
`/Users/username/mambaforge/envs/clip/lib/python3.10/site-packages` is where mine hides.
Change
```
if self._model is None:
self._model = SentenceTransformer('clip-ViT-B-32')
```
to point to your new model, like so:
```
def embed_batch(self, items):
# Embeds a mix of text strings and binary images
if self._model is None:
self._model = SentenceTransformer('/path/to/your/retrained-model')
```
The folder with your model should contain a pytorch_model.bin and config.json inside a subfolder called 0CLIP_Model. You will need the extra json files and so on from here [https://huggingface.co/sentence-transformers/clip-ViT-B-32/tree/main](https://huggingface.co/sentence-transformers/clip-ViT-B-32/tree/main) . You need all those .json files, arranged that way. And since you're not otherwise futzing with the basic CLIP-ness, it should be ok.
Once you create your embeddings, these will be in your ~Library/Application Support/io.datasette.llm folder.