Aquileo | nari-labs/Dia-1.6B · AMD Support?

nari-labs
/

Dia-1.6B

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

AMD Support?

#40

by Avaredra - opened May 4, 2025

Any chance of rocm support for AMD? I tried running the quick install, but the errors resulting indicate nvidia is required.

It works on AMD cards too, but you're basically on your own when it comes to installing the dependencies.

Create a virtualenv and activate it
Install the correct version of torch and torchaudio from here and then
Install the rest of the dependencies with pip install -e .

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment