update
This commit is contained in:
6
run_in_docker.sh
Normal file
6
run_in_docker.sh
Normal file
@@ -0,0 +1,6 @@
|
||||
#! /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'
|
||||
Reference in New Issue
Block a user