ailia-ai / ailia-ai/ailia-models

fix : latent-diffusion-txt2img require FP32

オープン
#1,030 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る
主要言語
Python
スター
2.4k
フォーク
365
平均マージ
23時間 2分
マージ済み PR(30日)
11

説明

[diffusion/latent-diffusion-txt2img](https://github.com/axinc-ai/ailia-models/tree/master/diffusion/latent-diffusion-txt2img) dosen't work correctly with FP16 accuracy. The FP16 result has no "playing guitar" elements.

The `transformer_attn.onnx`'s internal blob has `~1e+6` value that is transferred to +Inf.

## cuDNN(fp32)

![cudnn32 output](https://user-images.githubusercontent.com/43570021/217501500-77fe2027-a6d7-4172-89d6-176c03c6a757.png)

## cuDNN(fp16)

![cudnn16 1 2 13 output](https://user-images.githubusercontent.com/43570021/217501557-c8540cc1-b4a6-4d4e-a329-60360ac24315.png)

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。