Hugging Face's logo

Spaces:
Suburst
/
NJU_Final_Transformer_Tanslation
Sleeping

App Files Community
Suburst commited on
Commit
bda8bc6
·
verified ·
1 Parent(s): 00fa629

Update src/model.py

Browse files
Files changed (1) hide show
  1. src/model.py +1 -1
src/model.py CHANGED
@@ -5,7 +5,7 @@ from loguru import logger
5
  import os
6
 
7
 
8
- DEVICE = torch.device("cuda:6" if torch.cuda.is_available() else "cpu")
9
 
10
  def get_sinusoid_encoding_table(max_len, d_model):
11
  pos_encoding = torch.zeros(max_len, d_model)
 
5
  import os
6
 
7
 
8
+ DEVICE = torch.device("cpu")
9
 
10
  def get_sinusoid_encoding_table(max_len, d_model):
11
  pos_encoding = torch.zeros(max_len, d_model)