dexmal/opendm

$ git clone https://github.com/dexmal/opendm.git
  cd opendm
  
  docker run -it --rm --gpus all --network host \
    --name opendm \
    --shm-size=16g \
    -v "$PWD":/app/opendm \
    -w /app/opendm \
    dexmal/opendm:latest /bin/bash

$ conda activate opendm
  pip install -e .

$ conda create -n opendm python=3.10 -y
  conda activate opendm
  pip install torch torchvision \
    --index-url https://download.pytorch.org/whl/cu128
  pip install ninja packaging
  MAX_JOBS=2 pip install flash-attn --no-build-isolation
  pip install -e .

$ pip install -e ".[fast-infer]"

$ huggingface-cli download Dexmal/DM05 --local-dir ./checkpoints/DM05

$ script/dm05_launcher.sh \
    --exp opendm/exp/dm05_exp.py \
    --task inference \
    --model-config.model-name-or-path ./checkpoints/DM05 \
    --model-config.chunk-size 50 \
    --inference-config.output-action-dim 14 \
    --inference-config.image-prompts "Head" "Left wrist" "Right wrist" \
    --inference-config.port 7891