SteganographywithGPT-2

Sleeping

Xinyoumeng233hu commited on May 14, 2023

Commit

c864132

1 Parent(s): 13010a8

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -66,7 +66,7 @@ def get_model(seed=1234, model_name='gpt2'):
     np.random.seed(seed)
     torch.random.manual_seed(seed)
     torch.cuda.manual_seed(seed)
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     enc = GPT2Tokenizer.from_pretrained(model_name)
     enc.unk_token = None

     np.random.seed(seed)
     torch.random.manual_seed(seed)
     torch.cuda.manual_seed(seed)
+    device = torch.device("cpu")
     enc = GPT2Tokenizer.from_pretrained(model_name)
     enc.unk_token = None