From 9201ec5cb6aedef1e9cd83b2d51eecc894eee856 Mon Sep 17 00:00:00 2001
From: AccidentallyOnPurpose
 <4935931+AccidentallyOnPurpose@users.noreply.github.com4935931+AccidentallyOnPurpose@users.noreply.github.com>
Date: Mon, 23 Dec 2019 22:35:30 +0800
Subject: [PATCH] missing a not, was forcing cpu

---
 gpt2generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpt2generator.py b/gpt2generator.py
index 132260b..0ad8812 100644
--- a/gpt2generator.py
+++ b/gpt2generator.py
@@ -7,7 +7,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 from getconfig import settings, logger
 from story.utils import cut_trailing_sentence
 
-CPU = torch.cuda.is_available() and not settings.getboolean('force-cpu')
+CPU = not torch.cuda.is_available() and not settings.getboolean('force-cpu')
 
 # warnings.filterwarnings("ignore")
 MODEL_CLASSES = {