From ef45ad99d0916d2b7c6f7f472129566c535efbdc Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Fri, 27 Sep 2019 14:23:18 -0600
Subject: [PATCH] update

---
 generator/ctrl/ctrl_generator.py | 2 +-
 generator/ctrl/download_model.sh | 9 ++-------
 2 files changed, 3 insertions(+), 8 deletions(-)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 1341897..a5d5288 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -23,7 +23,7 @@ class CTRLGenerator():
     def __init__(self, control_code="Writing Text: ", generate_num=64, temperature=0.5, topk=40):
 
         self.generate_num=generate_num
-        model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"
+        model_dir = "generator/ctrl/model/seqlen512_v1.ckpt/"
         self.control_code = control_code
         vocab_file = 'generator/ctrl/model/vocab'
         code_file = 'generator/ctrl/model/codes'
diff --git a/generator/ctrl/download_model.sh b/generator/ctrl/download_model.sh
index 0b757ad..e002f85 100755
--- a/generator/ctrl/download_model.sh
+++ b/generator/ctrl/download_model.sh
@@ -2,12 +2,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
-    URL="gs://sf-ctrl/seqlen512_v1.ckpt/"
-else
-    URL="gs://sf-ctrl/seqlen256_v1.ckpt/"
-fi
+
+URL="gs://sf-ctrl/seqlen512_v1.ckpt/"
 
 gsutil -m cp -r "$URL" model