entry

app.py

@@ -234,7 +234,10 @@ def main():
 
     # Prepare the conditioning
     cond_tokens, cond_str_tokens = description2tokens(description, metadata.word2id , cfg)
-    cond_tokens = torch.tensor(cond_tokens).long().cuda()
+    if is_cuda:
+        cond_tokens = torch.tensor(cond_tokens).long().cuda()
+    else:
+        cond_tokens = torch.tensor(cond_tokens).long()
     if pointer_words is not None:
         numberical_conditioning = [float(description["cost_to_pointer"][key]) for key in pointer_words if key in description["cost_to_pointer"]]
     else: