wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-10 11:40:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

made it download the pytorch model v5 from torrent and rename

Browse Source
This commit is contained in:
AccidentallyOnPurpose
2019-12-24 09:50:22 +00:00
committed by GitHub
parent e331d0d621
commit ccb4526e50
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
download_model.sh
+5 -5
View File
@@ -3,13 +3,13 @@ cd "$(dirname "${0}")"
BASE_DIR="$(pwd)"
BASE_DIR="$(pwd)"
MODELS_DIRECTORY=generator/gpt2/models
MODELS_DIRECTORY=models
MODEL_VERSION=model_v5
MODEL_DIRECTORY="${MODELS_DIRECTORY}"
MODEL_NAME=model-550
MODEL_TORRENT_URL="https://github.com/AIDungeon/AIDungeon/files/3935881/model_v5.torrent.zip"
MODEL_NAME=pytorch-gpt2-xl-aid2-v5
MODEL_TORRENT_URL="https://raw.githubusercontent.com/AccidentallyOnPurpose/pytorch-AIDungeon/f692e39d84b21d39da9819142165a05a03030892/generator/gpt2/models/model_v5_pytorch.torrent"
MODEL_TORRENT_BASENAME="$(basename "${MODEL_TORRENT_URL}")"
download_torrent() {
@@ -17,7 +17,6 @@ download_torrent() {
mkdir -p "${MODEL_DIRECTORY}"
cd "${MODEL_DIRECTORY}"
wget "${MODEL_TORRENT_URL}"
unzip "${MODEL_TORRENT_BASENAME}"
which aria2c > /dev/null
if [ $? == 0 ]; then
echo -e "\n\n==========================================="
@@ -32,8 +31,9 @@ download_torrent() {
--seed-time=0 \
--summary-interval=15 \
--disable-ipv6 \
"${MODEL_TORRENT_BASENAME%.*}"
"${MODEL_TORRENT_BASENAME}"
echo "Download Complete!"
mv "${MODEL_TORRENT_BASENAME%.*}" $MODEL_NAME
fi
}