Somekindofa commited on
Commit
2f00c8b
·
1 Parent(s): dcfe987

Augmented Max input token

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = 20000
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