KB-VQA

Sleeping

m7mdal7aj commited on Feb 22, 2024

Commit

747fb93

verified ·

1 Parent(s): d31287c

Update my_model/captioner/captioning_config.py

Files changed (1) hide show

my_model/captioner/captioning_config.py CHANGED Viewed

@@ -6,7 +6,7 @@ PROMPT = "describe this image in details"
 MAX_IMAGE_SIZE = 1024
 MIN_LENGTH = 20
 MAX_NEW_TOKENS = 200
-MODEL_PATH = "Salesforce/instructblip-vicuna-7b"
 LOAD_IN_8BIT = True
 TORCH_DTYPE = torch.float16
 DEVICE_MAP = "auto"

 MAX_IMAGE_SIZE = 1024
 MIN_LENGTH = 20
 MAX_NEW_TOKENS = 200
+MODEL_PATH = "m7mdal7aj/captioner"
 LOAD_IN_8BIT = True
 TORCH_DTYPE = torch.float16
 DEVICE_MAP = "auto"