这是Boris Dayma的DALL·E 迷你。它已被剥离到进行推理所需的基本要素,并转换为PyTorch。要运行 PyTorch 模型,唯一的第三方依赖项是 numpy 和 torch。Flax用于转换权重(加载模型后可以保存),而wandb仅用于下载模型。
torch.save
目前需要7.3秒才能生成带有DALL·Colab上的PyTorch中的E Mega(在高RAM运行时中具有非消耗性模型)
运行以安装依赖项并下载预训练的模型。这些模型也可以在这里手动下载:VQGan,DALL·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