Merge pull request #1 from nisargjoshi10/master

fixed import error
oriondollar · Feb 5, 2021 · 6a0fb8d · 6a0fb8d
2 parents 417fc78 + a23a540
commit 6a0fb8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/parsers.py b/scripts/parsers.py
@@ -1,6 +1,6 @@
 import argparse
-from TransVAE.transvae.trans_models import TransVAE
-from TransVAE.transvae.rnn_models import RNN, RNNAttn
+from transvae.trans_models import TransVAE
+from transvae.rnn_models import RNN, RNNAttn
 
 def model_init(args, params={}):
     ### Model Name