stable-diffusion-xl-base-1.0 模型推理
可在線運(yùn)行的notebook示例:在kaggle網(wǎng)站搜索 stable-diffusion-xl-base-1.0
首先需要安裝依賴包:
接著加載模型:(注意這里設(shè)置16精度加載模型可以節(jié)省顯存,同時(shí)注意設(shè)置模型為gpu的cuda加速加快模型推理)
然后設(shè)置文本指令詞并生成你想要的圖片,最后保存圖片:
若你使用notebook編程環(huán)境,可以在notebook中查看圖片:
標(biāo)簽: