update

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

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

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" ]