Hugging Face's logo

Spaces:
stevengrove
/
YOLO-World
Running on T4

App Files Community
13
YOLO-World
File size: 1,199 Bytes 
186701e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
#!/bin/bash
DOCKER_IMAGE="mirrors.tencent.com/ronnysong_rd/fastdet:torch2.0.1-cuda11.7"

if [ ! -n "$DEBUG" ]; then
    COMMAND_PREFIX="pip3 install -e ."
else
    COMMAND_PREFIX="pip3 install -q -e third_party/mmengine;
                    pip3 install -q -e third_party/mmdetection;
                    pip3 install -q -e third_party/mmcv;
                    pip3 install -q -e third_party/mmyolo;
                    pip3 install -q -e ."
fi

sudo nvidia-docker run \
    --rm \
    -it \
    -e NVIDIA_VISIBLE_DEVICES=all \
    --env="DISPLAY" \
    --env="QT_X11_NO_MITSHM=1" \
    --volume="$HOME/.Xauthority:/root/.Xauthority:rw" \
    --shm-size=20gb \
    --network=host \
    -v /apdcephfs/:/apdcephfs/ \
    -v /apdcephfs_cq2/:/apdcephfs_cq2/ \
    -v /apdcephfs_cq3/:/apdcephfs_cq3/ \
    -v /data/:/data/ \
    -w $PWD \
    $DOCKER_IMAGE \
    bash -c "export TRANSFORMERS_CACHE=$PWD/work_dirs/.cache/transformers;
             export TORCH_HOME=$PWD/work_dirs/.cache/torch;
             export CLIP_CACHE=$PWD/work_dirs/.cache/clip;
             export HF_HOME=$PWD/work_dirs/.cache/hf;
             export TOKENIZERS_PARALLELISM=false;
             $COMMAND_PREFIX
             $*"