From aec90d195145b0787638fa8cabc64bda32aac828 Mon Sep 17 00:00:00 2001
From: root <nickwalton00@gmail.com>
Date: Fri, 1 Nov 2019 20:15:38 +0000
Subject: [PATCH] changedmodelpath

---
 generator/ctrl/ctrl_generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index eda359e..419d223 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -23,7 +23,7 @@ class CTRLGenerator():
     def __init__(self, control_code="Apocalypse ", generate_num=28, temperature=0.5, topk=40, nucleus_prob=0):
 
         self.generate_num=generate_num
-        model_dir = "generator/ctrl/training_utils/seqlen256_v1.ckpt/"
+        model_dir = "generator/ctrl/model/aidungeon2model/"
         self.control_code = control_code
         vocab_file = 'generator/ctrl/model/vocab'
         code_file = 'generator/ctrl/model/codes'