跳转到主要内容

未提供项目描述

项目描述

AutoCog

PyPI version

使用GPT4自动生成 predict.py 和 cog.yaml

安装

pip install autocog

使用方法

首先,使用 --ai-provider 指定您的AI提供商,并在环境变量中设置您的OpenAI/Anthropic API密钥

OPENAI_API_KEY=sk-...
ANTHROPIC_API_KEY=sk-...

在您想要cog-ify的仓库中运行

autocog

这将根据当前目录中的文件生成cog.yaml和predict.py。然后运行模型,如果运行失败,它将尝试修复错误并再次运行。默认情况下,它有5次修复尝试,这可以通过 --attempts 标志更改。

如果您的模型需要GPU来运行,您需要在GPU机器上运行AutoCog。

人工干预

有时AutoCog无法创建有效的Cog配置。在这些情况下,您(人类)必须介入并编辑cog.yaml和predict.py文件。

编辑完成后,通过再次运行 autocog 让AutoCog继续。如果您想从头开始重新创建 predict.pycog.yaml,请运行 autocog --initialize

默认情况下,AutoCog会猜测一个 cog predict 命令来运行模型。如果您想指定自己的predict命令,请使用 --predict-command 标志。

如果您想让AutoCog按照特定方向生成 cog.yamlpredict.py,可以使用 --tell 标志来提示GPT4

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

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解有关 安装包 的更多信息。

源分布

autocog-0.0.11.tar.gz (18.7 kB 查看哈希值)

上传时间 源代码

构建版本

autocog-0.0.11-py3-none-any.whl (22.6 kB 查看哈希值)

上传时间 Python 3

支持