Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA
Sleeping

App Files Community
2
m7mdal7aj commited on
Commit
d31287c
·
verified ·
1 Parent(s): a665e05

Update my_model/captioner/captioning_config.py

Browse files
Files changed (1) hide show
  1. my_model/captioner/captioning_config.py +1 -1
my_model/captioner/captioning_config.py CHANGED
@@ -5,7 +5,7 @@ MODEL_TYPE = "i_blip"
5
  PROMPT = "describe this image in details"
6
  MAX_IMAGE_SIZE = 1024
7
  MIN_LENGTH = 20
8
- MAX_NEW_TOKENS = 100
9
  MODEL_PATH = "Salesforce/instructblip-vicuna-7b"
10
  LOAD_IN_8BIT = True
11
  TORCH_DTYPE = torch.float16
 
5
  PROMPT = "describe this image in details"
6
  MAX_IMAGE_SIZE = 1024
7
  MIN_LENGTH = 20
8
+ MAX_NEW_TOKENS = 200
9
  MODEL_PATH = "Salesforce/instructblip-vicuna-7b"
10
  LOAD_IN_8BIT = True
11
  TORCH_DTYPE = torch.float16