From c1839bd367b1c01b6166b5f0bd817eb14e00e3e3 Mon Sep 17 00:00:00 2001
From: sangwon38383 <54392189+sangwon38383@users.noreply.github.com>
Date: Sat, 11 Apr 2020 12:13:26 +0900
Subject: [PATCH] Update models.py

data type error
---
 lang/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lang/models.py b/lang/models.py
index 3bc7cac..3f4ad6c 100644
--- a/lang/models.py
+++ b/lang/models.py
@@ -273,7 +273,7 @@ def generate(autoencoder, gan_gen, z, vocab, sample, maxlen):
     """
     if type(z) == Variable:
         noise = z
-    elif type(z) == torch.FloatTensor or type(z) == torch.cuda.FloatTensor:
+    elif type(z) == torch.Tensor or type(z) == torch.cuda.Tensor:
         noise = Variable(z, volatile=True)
     elif type(z) == np.ndarray:
         noise = Variable(torch.from_numpy(z).float(), volatile=True)