https://github.com/justinrubek/generation-toolkit
A collection of tools for working with generative models. This is a work in progress as I need more tools for my own use cases
https://github.com/justinrubek/generation-toolkit
diffusion flake gpt image-generation libtorch llm nix nix-flake openai rust stable-diffusion text2image
Last synced: about 2 months ago
JSON representation
A collection of tools for working with generative models. This is a work in progress as I need more tools for my own use cases
- Host: GitHub
- URL: https://github.com/justinrubek/generation-toolkit
- Owner: justinrubek
- Created: 2023-06-05T22:51:44.000Z (about 3 years ago)
- Default Branch: main
- Last Pushed: 2023-08-25T23:11:16.000Z (almost 3 years ago)
- Last Synced: 2025-10-26T02:45:54.385Z (8 months ago)
- Topics: diffusion, flake, gpt, image-generation, libtorch, llm, nix, nix-flake, openai, rust, stable-diffusion, text2image
- Language: Rust
- Homepage:
- Size: 121 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- Changelog: CHANGELOG.md
Awesome Lists containing this project
README
# generation-toolkit
This is a work in progress collection of utilities for generating text/images.
This includes calling the OpenAI API as well as some local generation when possible.
This started as a place where I could prototype using OpenAI APIs and is expanding as I see fit.
## utilities
### stable diffusion
A nix flake package has been exposed that will help you get the weights needed to run stable diffusion.
You can use nix to create the symlink that the cli will expect for stable diffusion commands: `nix build .#stable-diffusion2-1 -o models`.
After doing this you may generate an image: `nix run github:justinrubek/generation-toolkit#cli stable-diff generate --prompt "A crab holding a flaming torch"`
### openai
You can generate a string that contains the code from a given directory.
All files will be included with a separating line and the file name at the top.
In order to ignore files, you can include a `.gptignore` file which will filter out files.
This can be included in any directory (thanks to the `ignore` crate) and will be applied similarly to a .gitignore.
To use it, invoke the cli with a path to walk: `nix run github:justinrubek/generation-toolkit#cli util generate-prompt .`