clip-retrieval VS stable-diffusion

The tool clip-retreival lets you filter those 400 million images to whatever subsets you're interested in --- for example, 10,000 images of (mostly) handwriting.

Hehe, well you know, PR welcome, the front end is 500 lines https://github.com/rom1504/clip-retrieval/blob/main/front/sr...

Other people have done a few alternate front ends already

This one is meant to be functional, but could sure be made prettier

This might work: https://github.com/rom1504/clip-retrieval .

HaveIBeenTrained uses clip retrieval to search the Laion-5B and Laion-400M image datasets. These are currently the largest public text-to-image datsets, and they are used to train models like Stable Diffusion, Imagen, among many others.

Done https://github.com/rom1504/clip-retrieval/commit/53e3383f58b...

Using clip for searching is better than direct text indexing for a variety of reasons but here for example because it matches better what stable diffusion sees

Based on OpenAI's CLIP and the clip-retrieval library (https://github.com/rom1504/clip-retrieval), I've built an end-to-end demo for a semantic and similarity image search engine. It's incredibly powerful for finding similar images amongst large image datasets, or just submitting text/natural language queries and finding the most relevant images in your dataset. Really useful tool for introspection into large datasets before annotation or ML work begins. This could potentially be used to filter or downsize your datasets by several orders of magnitude and make annotation and ML work easier and less costly.

Checkout the demo here:

http://ec2-52-39-251-116.us-west-2.compute.amazonaws.com/

And you can checkout our website or email me for updates and email list, etc.:

https://machineperception.co

Assuming you have an NVIDIA GPU, you can build a semantic search engine by indexing CLIP embeds (image or text).

https://github.com/rom1504/clip-retrieval

Stable Diffusion: It is based on a kind of diffusion model called a latent diffusion model, which is trained to remove noise from images in an iterative process. It is one of the first text-to-image models that can run on consumer hardware and has its code and model weights publicly available.

Which is a 1-click install of Stable Diffusion with an alternative web interface. You can choose a different approach but this one is pretty simple and I am new to this stuff.

an invisible watermarking of the outputs, to help viewers identify the images as machine-generated.

It sounds like you just want to set up Stable Diffusion to run locally. I don't think your computer's specs will be able to do it. You need a graphics card with a decent amount of VRAM. Stable diffusion is in Python as is almost every AI open source project I've seen. If you can get your hands on a system with an Nvidia RTX card with as much VRAM as possible, you're in business. I have an RTX 3060 with 12 gigs of VRAM and I can run stable diffusion and a whole variety of open source LLMs as well as other projects like face swap, Roop, tortoise TTS, sadtalker, etc...

Use specific GPU on multi GPU systems · Issue #87 · CompVis/stable-diffusion · GitHub

I dealt with this in one of my posts. At least SD 1.1 till 1.5 are all trained on a batch size of 2048. The version pretty much everyone uses (1.5) is first pretrained at a resolution of 256x256 for 237K steps on laion2B-en, at the end of those training steps it will have seen roughly 500M images in laion2B-en. After that it is pre-trained for 194K steps on laion-high-resolution at a resolution of 512x512, which is a subset of 170M images from laion5B. Finally it is trained for 1.110K steps on LAION aesthetic v2 5+. This is easily verified by taking a glance at the model card of SD 1.5. Though that one doesn't specify for part of the training exactly which aesthetic set was used for part of the training, for that you have to look at the CompVis github repo. Thus at the end of it all both the most recent images and the majority of images will have come from LAION aesthetic v2 5+ (seeing every image approx 4 times). Realistically a lot of the weights obtained from pretraining on 2B will have been lost, and only provided a good starting point for the weights.

stable diffusion · CompVis/stable-diffusion@2ff270f · GitHub