最近在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