6 lines
690 B
Bash
6 lines
690 B
Bash
#! /usr/bin/env bash
|
|
image=r200_8f-diffuser:v0.21.4
|
|
docker run -e TZ=Asia/Shanghai -v /mnt/data/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 \
|
|
--security-opt=seccomp=unconfined --cap-add=SYS_PTRACE --cap-add=SYS_ADMIN --device /dev/fuse --shm-size=32g --ulimit=memlock=-1 --ulimit=nofile=120000 --ulimit=stack=67108864 --device=/dev/xpu1:/dev/xpu1 --device=/dev/xpuctrl:/dev/xpuctrl \
|
|
$image \
|
|
/bin/bash -c 'source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && python3 run_diffuser_test_r200_8f.py --model /workspace/stable-diffusion-v1-5 --json dataset.json --results result.json --outdir output --device cuda --dtype fp16' |