ailia-ai / ailia-ai/ailia-models

fix : latent-diffusion-txt2img require FP32

Aperta
#1,030 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
2.4k
Fork
365
Merge medio
3g 17h
PR unite (30g)
13

Descrizione

[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)

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.