mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-30 12:50:18 +08:00
missing a not, was forcing cpu
This commit is contained in:
1 parent
7489b69955
commit
9201ec5cb6
1 file changed
+1
-1
+1
-1
@@ -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 = {
|
||||
|
||||
Reference in new issue
Block a user