Somekindofa
commited on
Commit
·
2f00c8b
1
Parent(s):
dcfe987
Augmented Max input token
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
|
16 |
MAX_MAX_NEW_TOKENS = 4096
|
17 |
DEFAULT_MAX_NEW_TOKENS = 2048
|
18 |
# MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
19 |
-
MAX_INPUT_TOKEN_LENGTH =
|
20 |
|
21 |
DEFAULT_USER_QUERY = '''
|
22 |
@prefix : <urn:webprotege:ontology:7272b2af-011f-4d40-8519-9fc3f830442e#> .
|
@@ -867,7 +867,7 @@ def generate(
|
|
867 |
|
868 |
|
869 |
streamer = TextIteratorStreamer(tokenizer,
|
870 |
-
timeout=60.0,
|
871 |
skip_prompt=True,
|
872 |
skip_special_tokens=True)
|
873 |
|
|
|
16 |
MAX_MAX_NEW_TOKENS = 4096
|
17 |
DEFAULT_MAX_NEW_TOKENS = 2048
|
18 |
# MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
19 |
+
MAX_INPUT_TOKEN_LENGTH = 100000
|
20 |
|
21 |
DEFAULT_USER_QUERY = '''
|
22 |
@prefix : <urn:webprotege:ontology:7272b2af-011f-4d40-8519-9fc3f830442e#> .
|
|
|
867 |
|
868 |
|
869 |
streamer = TextIteratorStreamer(tokenizer,
|
870 |
+
timeout=2*60.0,
|
871 |
skip_prompt=True,
|
872 |
skip_special_tokens=True)
|
873 |
|