gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

模型訓練

Browse Source
This commit is contained in:
桜華
2023-05-28 00:44:13 +08:00
parent 0966a3c83e
commit ed7e09e736
5 changed files with 92 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
test.sh
View File

@@ -53,19 +53,19 @@ response=$(curl -X PATCH -H "Content-Type: application/json" -d '{"images":["htt
[[ ${response: -3} -eq 200 ]] && { echo "修改數據集成功: ${response%???}"; } || exit_service "修改數據集失敗: ${response%???}"
# 訓練模型 (POST /api/models)
response=$(curl -X POST -H "Content-Type: application/json" -d '{"name":"test","type":"lora","trigger_words":"miao~","base_model":"sd1.5","epochs":20,"description":"test","dataset_id":'$dataset_id'}' -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/models)
[[ ${response: -3} -eq 200 ]] && { echo "訓練模型任務已創建: ${response%???}"; } || exit_service "訓練模型任務創建失敗: ${response%???}"
## 訓練模型 (POST /api/models)
#response=$(curl -X POST -H "Content-Type: application/json" -d '{"name":"test","type":"dreambooth","trigger_words":"miao~","base_model":"sd1.5","epochs":20,"description":"test","dataset_id":'$dataset_id'}' -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/models)
#[[ ${response: -3} -eq 200 ]] && { echo "訓練模型任務已創建: ${response%???}"; } || exit_service "訓練模型任務創建失敗: ${response%???}"
#
#
## 取模型id的值, 值爲 int
#model_id=$(echo "${response%???}" | grep -o '"id": [0-9]*' | awk '{print $2}')
#echo "model_id: $model_id"
#
# 取模型id的值, 值爲 int
model_id=$(echo "${response%???}" | grep -o '"id": [0-9]*' | awk '{print $2}')
echo "model_id: $model_id"
## 模型列表 (GET /api/models)
# response=$(curl -X GET -H "Content-Type: application/json" -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/models)
# [[ ${response: -3} -eq 200 ]] && { echo "獲取模型列表成功: ${response%???}"; } || exit_service "獲取模型列表失敗: ${response%???}"
# 模型列表 (GET /api/models)
response=$(curl -X GET -H "Content-Type: application/json" -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/models)
[[ ${response: -3} -eq 200 ]] && { echo "獲取模型列表成功: ${response%???}"; } || exit_service "獲取模型列表失敗: ${response%???}"
## 獲取模型訓練進度 (GET /api/models/:id)