From c2857098d09fc5c07adbe41699d8244b1a7be4d7 Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Wed, 18 Sep 2019 12:57:28 -0600
Subject: [PATCH] updated scripts

---
 generator/ctrl/download_model.sh | 2 +-
 generator/ctrl/install_ctrl.sh   | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/generator/ctrl/download_model.sh b/generator/ctrl/download_model.sh
index ce771bc..e42bf19 100755
--- a/generator/ctrl/download_model.sh
+++ b/generator/ctrl/download_model.sh
@@ -6,4 +6,4 @@ else
     URL="gs://sf-ctrl/seqlen256_v1.ckpt/"
 fi
 
-gsutil -m cp -r "$URL" .
+gsutil -m cp -r "$URL" model
diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh
index 7cc5a92..349032d 100755
--- a/generator/ctrl/install_ctrl.sh
+++ b/generator/ctrl/install_ctrl.sh
@@ -16,6 +16,7 @@ cd fastBPE
 python setup.py install
 cd ..
 
+pip install tensorflow-gpu==1.14
 # Download the 512-length model if specified, 256-length otherwise
 #if [ "$1" = "512" ]
 #then