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