abdfajar707 commited on
Commit
c6c8b82
·
verified ·
1 Parent(s): 196ef8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,7 +7,6 @@ from typing import Iterator
7
  from transformers import (
8
  AutoModelForCausalLM,
9
  BitsAndBytesConfig,
10
- GemmaTokenizerFast,
11
  TextIteratorStreamer,
12
  )
13
 
@@ -37,7 +36,7 @@ if True:
37
 
38
  MAX_MAX_NEW_TOKENS = 2048
39
  DEFAULT_MAX_NEW_TOKENS = 1024
40
- MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
41
 
42
 
43
  model.config.sliding_window = 4096
 
7
  from transformers import (
8
  AutoModelForCausalLM,
9
  BitsAndBytesConfig,
 
10
  TextIteratorStreamer,
11
  )
12
 
 
36
 
37
  MAX_MAX_NEW_TOKENS = 2048
38
  DEFAULT_MAX_NEW_TOKENS = 1024
39
+ MAX_INPUT_TOKEN_LENGTH = 4096 #int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
40
 
41
 
42
  model.config.sliding_window = 4096