#!/bin/bash
TARGET_DIR=gpu-base-image-test
docker run --rm --platform=linux/amd64 --gpus all -v ./$TARGET_DIR:/workspace --workdir /workspace/gpt2 $1 python infer.py
