Hugging Face's logo

Spaces:
prithivMLmods
/
Multimodal-OCR2
Running on Zero

App Files Community
7
prithivMLmods commited on
Commit
9c5d023
·
verified ·
1 Parent(s): 6ba01bd

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -107,8 +107,8 @@ css = """
107
  """
108
 
109
  # Constants for text generation
110
- MAX_MAX_NEW_TOKENS = 5120
111
- DEFAULT_MAX_NEW_TOKENS = 3072
112
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
113
 
114
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
107
  """
108
 
109
  # Constants for text generation
110
+ MAX_MAX_NEW_TOKENS = 4096
111
+ DEFAULT_MAX_NEW_TOKENS = 2048
112
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
113
 
114
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")