min-dalle - min(DALL·E) 是 PyTorch 中 DALL·E Mini 的最小实现

Created at: 2022-06-27 23:53:59
Language: Python
License: MIT

最小值(DALL·E)

在科拉布中打开   复制

这是Boris Dayma的DALL·E 迷你。它已被剥离到进行推理所需的基本要素,并转换为PyTorch。要运行 PyTorch 模型,唯一的第三方依赖项是 numpy 和 torch。Flax用于转换权重(加载模型后可以保存),而wandb仅用于下载模型。

torch.save

目前需要7.3秒才能生成带有DALL·Colab上的PyTorch中的E Mega(在高RAM运行时中具有非消耗性模型)

设置

运行以安装依赖项并下载预训练的模型。这些模型也可以在这里手动下载:VQGanDALL·E迷你达力·E 兆丰

sh setup.sh

用法

使用 python 脚本从命令行生成图像。注意:命令行脚本每次加载模型和参数。要加载一个模型一次并生成多次,请初始化 或 ,然后使用一些文本和种子进行调用。有关示例,请参阅 colab。

image_from_text.py
MinDalleTorch
MinDalleFlax
generate_image

例子

python image_from_text.py --text='artificial intelligence' --torch

外人

python image_from_text.py --text='a comfy chair that looks like an avocado' --torch --mega --seed=10

鳄梨扶手椅

python image_from_text.py --text='court sketch of godzilla on trial' --mega --seed=100

哥斯拉审判