最近在kaggle中发现了一位大佬写的用tpu进行ai画画,https://www.kaggle.com/code/loseman/novelai-tpu

多张图生成的速度比GPU快不少,单图速度和GPU差不多。

修改了一下大佬的笔记本,使得可以将生成的图片输出至目录,并压缩方便下载。

kaggle:https://www.kaggle.com/code/sakuraxiaowen/novelai-aidraw-tup

colab:https://colab.research.google.com/drive/1Qrn3B0uv7mQmngYiCptfwlYdtfm9jcPz?usp=sharing