Skip to main content

No project description provided

Project description

AutoCog

PyPI version

Generate predict.py and cog.yaml automatically using GPT4

Install

pip install autocog

Usage

First, set your OpenAI API key in an environment variable

OPENAI_API_KEY=sk-...

In the repo you want to cog-ify, run

autocog

This will generate a cog.yaml and predict.py based on the files in the current directory. It will then run the model and if it fails to run, it will attempt to fix the error and run it again. By default it has 5 attempts to fix it, which can be changed with the --attempts flag.

If your model needs a GPU to run, you need to run AutoCog on a GPU machine.

Human in the loop

Sometimes AutoCog fails to create a working Cog configuration. In those cases you, the human, have to step in and edit the cog.yaml and predict.py files.

Once you have edited them, let AutoCog continue by running autocog again. If you'd like to recreate predict.py and cog.yaml from scratch, run autocog --initialize.

By default, AutoCog will guess a cog predict command to run the model. If you want to specify your own predict command, use the --predict-command flag.

If you want AutoCog to take the generation of cog.yaml and predict.py in a specific direction, you can use the --tell flag to prompt GPT4:

autocog --tell="Add inputs to allow for inpainting"

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

autocog-0.0.5.tar.gz (9.0 kB view hashes)

Uploaded Source

Built Distribution

autocog-0.0.5-py3-none-any.whl (9.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page