ONNX 模型库
返回模型

说明文档

这是 PFG 的权重文件存放处。可以直接 clone pfg 并在 generate.py 中使用,也可以在 Colab 中快速试用。

wd-v1-4-vit-tagger-v2-last-pooling-layer.onnx

这是输出最后的池化层的 onnx 文件。由 wd-v1-4-vit-tagger-v2 的 onnx 文件创建而成。

wd14-n10

这是使用 waifu-diffusion 训练的模型。token 数量为 10 个。 SD2 系列的其他模型(无论是否涉及 v_pred)也可能在一定程度上使用。 训练数据为 61 万张,全部包含 1girl,因此可能无法应对多个角色。 此外,版权角色可能难以还原。毕竟原模型本身也难以还原。想要生成喜欢的角色的人,请针对该角色进行训练。

生成示例:

※出于版权考虑,输入的图像是用 waifu-diffusion 创建的。因此还原度比普通图像更高。基本上 waifu-diffusion 无法生成的图像,使用此方法也无法生成。

image image

wd15beta2-n10

这是从 wd14-n10 恢复,使用相同数据集针对 waifu-diffusion-1-5-beta2 训练的模型。

Licence

此文件无法单独运行,请遵守所使用的预训练模型的许可证。

furusu/PFG

作者 furusu

↓ 0 ♥ 21

创建时间: 2023-02-18 00:41:46+00:00

更新时间: 2023-03-28 00:39:43+00:00

在 Hugging Face 上查看

文件 (8)

.gitattributes
README.md
a1.png
a2.png
pfg-novel-n10.pt
pfg-wd14-n10.pt
pfg-wd15beta2-n10.pt
wd-v1-4-vit-tagger-v2-last-pooling-layer.onnx ONNX