From fa7104f2cffd0d4db4313ee7572b8d94c6f53325 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 11:12:35 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 2 -- generator/ctrl/download_model.sh | 4 ++++ generator/ctrl/install_ctrl.sh | 8 +------- 3 files changed, 5 insertions(+), 9 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index c2cda8a..f2c9e9b 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -259,7 +259,5 @@ class CTRLGenerator(): result = tokens_generated_so_far[prompt_length:] - except: - print("Error in generation") return result diff --git a/generator/ctrl/download_model.sh b/generator/ctrl/download_model.sh index e42bf19..0b757ad 100755 --- a/generator/ctrl/download_model.sh +++ b/generator/ctrl/download_model.sh @@ -1,3 +1,7 @@ +curl https://sdk.cloud.google.com | bash +exec -l $SHELL +gcloud init + # Download the 512-length model if specified, 256-length otherwise if [ "$1" = "512" ] then diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh index 99c5d2d..4a66e63 100755 --- a/generator/ctrl/install_ctrl.sh +++ b/generator/ctrl/install_ctrl.sh @@ -15,13 +15,7 @@ patch -b "$FILE" estimator.patch git clone https://github.com/glample/fastBPE.git cd fastBPE python setup.py install -cd .. - -curl https://sdk.cloud.google.com | bash -exec -l $SHELL -gcloud init -export CUDA_VISIBLE_DEVICES=0 -cd .. +cd ../.. # Download the 512-length model if specified, 256-length otherwise #if [ "$1" = "512" ]