Sign inSign up

amdsiloai/pytorch-xdit

By amdsiloai

•Updated 10 days ago

https://rocm.docs.amd.com/en/latest/how-to/rocm-for-ai/inference/xdit-diffusion-inference.html

Image
0

7.5K

amdsiloai/pytorch-xdit repository overview

⁠PyTorch xDiT Images

Preview ROCm-based images for running models using xDiT⁠.

For detailed instructions, see rocm-xdit-diffusion-inference⁠

For stable releases, see rocm/pytorch-xdit⁠

⁠Usage

Example docker run command:

docker pull amdsiloai/pytorch-xdit
docker run \
    -it --rm \
    --cap-add=SYS_PTRACE \
    --security-opt seccomp=unconfined \
    --user root \
    --device=/dev/kfd \
    --device=/dev/dri \
    --group-add video \
    --ipc=host \
    --network host \
    --privileged \
    --shm-size 128G \
    --name pytorch-xdit \
    -e CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7  \
    amdsiloai/pytorch-xdit

Tag summary

Content type

Image

Digest

sha256:1f27378ce…

Size

6.5 GB

Last updated

10 days ago

docker pull amdsiloai/pytorch-xdit