wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-09 11:13:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

missing a not, was forcing cpu

Browse Source
This commit is contained in:
AccidentallyOnPurpose
2019-12-23 22:35:30 +08:00
parent 7489b69955
commit 9201ec5cb6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gpt2generator.py
+1 -1
View File
@@ -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 = {