Spaces:
Sleeping
Sleeping
Commit
·
baa5dba
1
Parent(s):
496ea04
update
Browse files
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 |
|