Update my_model/captioner/captioning_config.py
Browse files
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 =
|
| 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
|