DEV Community

Anuroop
Anuroop

Posted on

I made a python package to store AI generated images along with their context (prompt, model info, etc)

Hey y'all, I made a python package that takes an image, it's context like the prompt, model name, hardware info, generation settings (e.g sampler settings),etc and also has 2 distinctive features:

  1. Can store Latent tensors
  2. Has a environment chunk that screates canonical strings of the e verisoning and stuff of critical libraries and hardware (eg pytorch, gpu drivers, OS, CPU cores etc. ) and hashes them and uses it to detect environment drift!

Check the repo and feel free to star it!!
https://github.com/AnuroopVJ/gen5

Top comments (0)