From da24f6b56d8fcae4a37fb09caa83806bd99e8c86 Mon Sep 17 00:00:00 2001
From: AccidentallyOnPurpose
 <4935931+AccidentallyOnPurpose@users.noreply.github.com4935931+AccidentallyOnPurpose@users.noreply.github.com>
Date: Tue, 24 Dec 2019 15:25:19 +0800
Subject: [PATCH] try to download new torrent

---
 download_model.sh | 5 +++--
 requirements.txt  | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/download_model.sh b/download_model.sh
index c8b5f29..4290bb3 100644
--- a/download_model.sh
+++ b/download_model.sh
@@ -8,8 +8,8 @@ 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://github.com/AccidentallyOnPurpose/pytorch-AIDungeon/blob/f692e39d84/generator/gpt2/models/model_v5_pytorch.torrent?raw=true"
 MODEL_TORRENT_BASENAME="$(basename "${MODEL_TORRENT_URL}")"
 
 download_torrent() {
@@ -34,6 +34,7 @@ download_torrent() {
       --disable-ipv6 \
       "${MODEL_TORRENT_BASENAME%.*}"
     echo "Download Complete!"
+    mv "${MODEL_TORRENT_BASENAME%.*}" $MODEL_NAME
     fi
 }
 
diff --git a/requirements.txt b/requirements.txt
index 110bc03..fdeb997 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 torch
 pyjarowinkler
-transformers
+transformers==2.3.0