hoduyquocbao commited on
Commit
baa5dba
·
1 Parent(s): 496ea04
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -38,6 +38,8 @@ MAX_MAX_NEW_TOKENS: int = 8**1024**4 # Số token tối đa có thể tạo ra
38
  DEFAULT_MAX_NEW_TOKENS: int = 8**1024**4 # Số token tạo ra mặc định
39
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "128000")) # Độ dài token tối đa cho đầu vào
40
 
 
 
41
  # Xác định thiết bị sử dụng (GPU nếu có, ngược lại CPU)
42
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
43
 
 
38
  DEFAULT_MAX_NEW_TOKENS: int = 8**1024**4 # Số token tạo ra mặc định
39
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "128000")) # Độ dài token tối đa cho đầu vào
40
 
41
+ print(MAX_INPUT_TOKEN_LENGTH)
42
+
43
  # Xác định thiết bị sử dụng (GPU nếu có, ngược lại CPU)
44
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
45