Просмотр исходного кода

添加运行本地目录环境

Samien 1 год назад
Родитель
Сommit
6eafaa106b
2 измененных файлов с 28 добавлено и 1 удалено
  1. 2 1
      .gitignore
  2. 26 0
      run_rwkv.sh

+ 2 - 1
.gitignore

@@ -34,5 +34,6 @@ convert_safetensors.py
 llms/convert_rwkv.py
 
 .DS_Store
-.vscode/launch.json
+.vscode
 run_rwkv_env.sh
+.idea

+ 26 - 0
run_rwkv.sh

@@ -1,6 +1,32 @@
 #!/bin/bash
 PYTHON=""
 # python程序位置,可搭配一键包或是省去每次切换环境
+# If $venv_dir is "-", then disable venv support
+use_venv=1
+if [[ $venv_dir == "-" ]]; then
+  use_venv=0
+fi
+
+
+# python3 executable
+if [[ -z "${python_cmd}" ]]
+then
+    python_cmd="python3"
+fi
+
+# git executable
+if [[ -z "${GIT}" ]]
+then
+    export GIT="git"
+else
+    export GIT_PYTHON_GIT_EXECUTABLE="${GIT}"
+fi
+
+# python3 venv without trailing slash (defaults to ${install_dir}/${clone_dir}/venv)
+if [[ -z "${venv_dir}" ]] && [[ $use_venv -eq 1 ]]
+then
+    venv_dir="venv"
+fi
 
 while true
 do

粤ICP备19079148号