Blackwell 50xx support

[MaximeHoude]

Awesome project, thanks for sharing!

Unfortunately, the installation requirements are not compatible with the new Nvidia Blackwell 50xx architecture.

Compatibility will be much appreciated. Thanks again!

[aejion]

Thanks! Which library you encountered the issue with during installation? Could you provide the detailed installation error message?

[MaximeHoude]

Thanks for the fast reply. It's regarding torch==2.4.0. Blackwell uses Nightly. That being said, I just saw that it requires 80gb of vram, so currently useless on any Blackwell architecture.

[aejion]

You can try the quantized checkpoint, which may save video memory but could result in a decrease in generation quality.

[petermg]

You can try the quantized checkpoint, which may save video memory but could result in a decrease in generation quality.

This results in: