build:
  cmd: project run build
  script:
    - docker login --username=ddreamai1992 --password guoying196567 registry.cn-shanghai.aliyuncs.com/wangmengdi/rag
    - docker build -t registry.cn-shanghai.aliyuncs.com/wangmengdi/rag:c10 .
  deps: []
  outs: []
login:
  cmd: project run login
  script:
    - docker login --username=ddreamai1992 --password guoying196567 registry.cn-shanghai.aliyuncs.com/wangmengdi/carbook-rag
  deps: []
  outs: []
download-llm:
  cmd: project run download-llm
  script:
    - huggingface-cli download Qwen/Qwen-7B-Chat-Int8  --local-dir ./plms/Qwen/Qwen-7B-Chat-Int8
      --local-dir-use-symlinks False
  deps: []
  outs: []
push:
  cmd: project run push
  script:
    - docker login --username=ddreamai1992 --password guoying196567 registry.cn-shanghai.aliyuncs.com/wangmengdi/rag
    - docker push registry.cn-shanghai.aliyuncs.com/wangmengdi/rag:c10
  deps: []
  outs: []
pull:
  cmd: project run pull
  script:
    - docker pull registry.cn-shanghai.aliyuncs.com/wangmengdi/carbook-rag:0.1
  deps: []
  outs: []
download-chatglm:
  cmd: project run download-chatglm
  script:
    - huggingface-cli download THUDM/chatglm3-6b  --local-dir ./plms/chatglm3-6b --local-dir-use-symlinks
      False
  deps: []
  outs: []
download-qwen-int4:
  cmd: project run download-qwen-int4
  script:
    - huggingface-cli download Qwen/Qwen-7B-Chat-Int4  --local-dir ./plms/Qwen-7B-Chat-Int4
      --local-dir-use-symlinks False
  deps: []
  outs: []
download-bge:
  cmd: project run download-bge
  script:
    - huggingface-cli download BAAI/bge-large-zh-v1.5 --local-dir ./plms/bge-large-zh
      --local-dir-use-symlinks False
    - huggingface-cli download BAAI/bge-base-zh-v1.5 --local-dir ./plms/bge-base-zh
      --local-dir-use-symlinks False
    - huggingface-cli download BAAI/bge-reranker-large --local-dir ./plms/bge-reranker-large
      --local-dir-use-symlinks False
    - huggingface-cli download BAAI/bge-reranker-base --local-dir ./plms/bge-reranker-base
      --local-dir-use-symlinks False
  deps: []
  outs: []
test:
  cmd: project run test
  script:
    - sudo mkdir -p /tcdata
    - sudo cp assets/测试问题.json /tcdata/test_question.json
    - sudo cp assets/初赛训练数据集.pdf /tcdata/trainning_data.pdf
    - bash run.sh
  deps: []
  outs: []
download-qwen:
  cmd: project run download-qwen
  script:
    - huggingface-cli download Qwen/Qwen-7B-Chat  --local-dir ./plms/Qwen-7B-Chat
      --local-dir-use-symlinks False
  deps: []
  outs: []
prepare:
  cmd: project run prepare
  script:
    - sudo mkdir -p /tcdata
    - sudo cp assets/测试问题.json /tcdata/test_question.json
    - sudo cp assets/初赛训练数据集.pdf /tcdata/trainning_data.pdf
  deps: []
  outs: []
git-setup:
  cmd: project run git-setup
  script:
    - git config --global user.name wangmengdi
    - git config --global user.email 790990241@qq.com
  deps: []
  outs: []
download-baichuan:
  cmd: project run download-baichuan
  script:
    - huggingface-cli download baichuan-inc/Baichuan2-7B-Chat  --local-dir ./plms/Baichuan2-7B-Chat
      --local-dir-use-symlinks False
  deps: []
  outs: []
download-yi-6b:
  cmd: project run download-yi-6b
  script:
    - huggingface-cli download 01-ai/Yi-6B-Chat  --local-dir ./plms/Yi-6B-Chat --local-dir-use-symlinks
      False
  deps: []
  outs: []
download-internlm:
  cmd: project run download-internlm
  script:
    - huggingface-cli download internlm/internlm-chat-7b  --local-dir ./plms/internlm-chat-7b
      --local-dir-use-symlinks False
  deps: []
  outs: []
download-alpaca:
  cmd: project run download-alpaca
  script:
    - huggingface-cli download hfl/chinese-alpaca-2-7b  --local-dir ./plms/chinese-alpaca-2-7b
      --local-dir-use-symlinks False
  deps: []
  outs: []
test-docker:
  cmd: project run test-docker
  script:
    - docker run -it --gpus all  --name test registry.cn-shanghai.aliyuncs.com/wangmengdi/rag:c10  /bin/bash
    - docker start test
    - docker exec test mkdir /tcdata
    - docker cp ./assets/测试问题.json test:/tcdata/test_question.json
    - docker cp ./assets/初赛训练数据集.pdf test:/tcdata/trainning_data.pdf
    - docker exec -it test bash run.sh
    - docker stop test
    - docker rm test
  deps: []
  outs: []