EngineX-Iluvatar/enginex-mr_series-diffusers
15
1
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

mount current dir to workspace

Browse Source
This commit is contained in:
Zhang Hao
2025-08-28 17:51:11 +08:00
parent faf81388ba
commit 455848137a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
run_in_docker.sh
View File

@@ -1,3 +1,3 @@
#! /usr/bin/env bash
image=diffusers:v0.1
docker run -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=dev/iluvatar1:/dev/iluvatar0 $image python3 main.py --model "./stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
docker run -v `pwd`:/workspace -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=dev/iluvatar1:/dev/iluvatar0 $image python3 main.py --model "./stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16