FangYin Cheng
2 years ago
parent
506176edf6
commit
909bb263a4
@ -0,0 +1,68 @@ |
||||
@echo off |
||||
setlocal |
||||
|
||||
:: Get script location and set working directory |
||||
for %%i in (%0) do set SCRIPT_LOCATION=%%~dpi |
||||
cd %SCRIPT_LOCATION% |
||||
cd .. |
||||
cd .. |
||||
set WORK_DIR=%CD% |
||||
|
||||
:: Check if sqlite3 is installed |
||||
where sqlite3 >nul 2>nul |
||||
if %ERRORLEVEL% neq 0 ( |
||||
echo sqlite3 not found, please install sqlite3 |
||||
exit /b 1 |
||||
) |
||||
|
||||
:: Default file paths |
||||
set DEFAULT_DB_FILE=DB-GPT\pilot\data\default_sqlite.db |
||||
set DEFAULT_SQL_FILE=DB-GPT\docker\examples\sqls\*_sqlite.sql |
||||
set DB_FILE=%WORK_DIR%\pilot\data\default_sqlite.db |
||||
set SQL_FILE= |
||||
|
||||
:argLoop |
||||
if "%1"=="" goto argDone |
||||
if "%1"=="-d" goto setDBFile |
||||
if "%1"=="--db-file" goto setDBFile |
||||
if "%1"=="-f" goto setSQLFile |
||||
if "%1"=="--sql-file" goto setSQLFile |
||||
if "%1"=="-h" goto printUsage |
||||
if "%1"=="--help" goto printUsage |
||||
goto argError |
||||
|
||||
:setDBFile |
||||
shift |
||||
set DB_FILE=%1 |
||||
shift |
||||
goto argLoop |
||||
|
||||
:setSQLFile |
||||
shift |
||||
set SQL_FILE=%1 |
||||
shift |
||||
goto argLoop |
||||
|
||||
:argError |
||||
echo Invalid argument: %1 |
||||
goto printUsage |
||||
|
||||
:printUsage |
||||
echo USAGE: %0 [--db-file sqlite db file] [--sql-file sql file path to run] |
||||
echo [-d|--db-file sqlite db file path] default: %DEFAULT_DB_FILE% |
||||
echo [-f|--sql-file sqlite file to run] default: %DEFAULT_SQL_FILE% |
||||
echo [-h|--help] Usage message |
||||
exit /b 0 |
||||
|
||||
:argDone |
||||
|
||||
if "%SQL_FILE%"=="" ( |
||||
if not exist "%WORK_DIR%\pilot\data" mkdir "%WORK_DIR%\pilot\data" |
||||
for %%f in (%WORK_DIR%\docker\examples\sqls\*_sqlite.sql) do ( |
||||
echo execute sql file: %%f |
||||
sqlite3 "%DB_FILE%" < "%%f" |
||||
) |
||||
) else ( |
||||
echo Execute SQL file %SQL_FILE% |
||||
sqlite3 "%DB_FILE%" < "%SQL_FILE%" |
||||
) |
||||
@ -0,0 +1,69 @@ |
||||
#!/bin/bash |
||||
# Only support SQLite now |
||||
|
||||
SCRIPT_LOCATION=$0 |
||||
cd "$(dirname "$SCRIPT_LOCATION")" |
||||
WORK_DIR=$(pwd) |
||||
WORK_DIR="$WORK_DIR/../.." |
||||
|
||||
if ! command -v sqlite3 > /dev/null 2>&1 |
||||
then |
||||
echo "sqlite3 not found, please install sqlite3" |
||||
exit 1 |
||||
fi |
||||
|
||||
DEFAULT_DB_FILE="DB-GPT/pilot/data/default_sqlite.db" |
||||
DEFAULT_SQL_FILE="DB-GPT/docker/examples/sqls/*_sqlite.sql" |
||||
DB_FILE="$WORK_DIR/pilot/data/default_sqlite.db" |
||||
SQL_FILE="" |
||||
|
||||
usage () { |
||||
echo "USAGE: $0 [--db-file sqlite db file] [--sql-file sql file path to run]" |
||||
echo " [-d|--db-file sqlite db file path] default: ${DEFAULT_DB_FILE}" |
||||
echo " [-f|--sql-file sqlte file to run] default: ${DEFAULT_SQL_FILE}" |
||||
echo " [-h|--help] Usage message" |
||||
} |
||||
|
||||
while [[ $# -gt 0 ]]; do |
||||
key="$1" |
||||
case $key in |
||||
-d|--db-file) |
||||
DB_FILE="$2" |
||||
shift # past argument |
||||
shift # past value |
||||
;; |
||||
-f|--sql-file) |
||||
SQL_FILE="$2" |
||||
shift |
||||
shift |
||||
;; |
||||
-h|--help) |
||||
help="true" |
||||
shift |
||||
;; |
||||
*) |
||||
usage |
||||
exit 1 |
||||
;; |
||||
esac |
||||
done |
||||
|
||||
if [[ $help ]]; then |
||||
usage |
||||
exit 0 |
||||
fi |
||||
|
||||
if [ -n $SQL_FILE ];then |
||||
mkdir -p $WORK_DIR/pilot/data |
||||
for file in $WORK_DIR/docker/examples/sqls/*_sqlite.sql |
||||
do |
||||
echo "execute sql file: $file" |
||||
sqlite3 $DB_FILE < "$file" |
||||
done |
||||
|
||||
else |
||||
echo "Execute SQL file ${SQL_FILE}" |
||||
sqlite3 $DB_FILE < $SQL_FILE |
||||
fi |
||||
|
||||
|
||||
@ -0,0 +1,47 @@ |
||||
#!/bin/bash |
||||
|
||||
# Get cuda version by torch |
||||
CUDA_VERSION=`python -c "import torch;print(torch.version.cuda.replace('.', '') if torch.cuda.is_available() else '')"` |
||||
|
||||
DEVICE="cpu" |
||||
CPU_OPT="basic" |
||||
|
||||
if [ "${CUDA_VERSION}" = "" ]; then |
||||
echo "CUDA not support, use cpu version" |
||||
else |
||||
DEVICE="cu${CUDA_VERSION//./}" |
||||
echo "CUDA version: $CUDA_VERSION, download path: $DEVICE" |
||||
fi |
||||
|
||||
echo "Checking CPU support:" |
||||
CPU_SUPPORT=$(lscpu) |
||||
|
||||
echo "$CPU_SUPPORT" | grep -q "avx " |
||||
if [ $? -eq 0 ]; then |
||||
echo "CPU supports AVX." |
||||
# CPU_OPT="AVX" |
||||
# TODO AVX will failed on my cpu |
||||
else |
||||
echo "CPU does not support AVX." |
||||
fi |
||||
|
||||
echo "$CPU_SUPPORT" | grep -q "avx2" |
||||
if [ $? -eq 0 ]; then |
||||
echo "CPU supports AVX2." |
||||
CPU_OPT="AVX2" |
||||
else |
||||
echo "CPU does not support AVX2." |
||||
fi |
||||
|
||||
echo "$CPU_SUPPORT" | grep -q "avx512" |
||||
if [ $? -eq 0 ]; then |
||||
echo "CPU supports AVX512." |
||||
CPU_OPT="AVX512" |
||||
else |
||||
echo "CPU does not support AVX512." |
||||
fi |
||||
|
||||
EXTRA_INDEX_URL="https://jllllll.github.io/llama-cpp-python-cuBLAS-wheels/$CPU_OPT/$DEVICE" |
||||
|
||||
echo "install llama-cpp-python from --extra-index-url ${EXTRA_INDEX_URL}" |
||||
python -m pip install llama-cpp-python --force-reinstall --no-cache --prefer-binary --extra-index-url=$EXTRA_INDEX_URL |
||||
Loading…
Reference in new issue