Update vlm_inference.py
Browse files- vlm_inference.py +1 -1
vlm_inference.py
CHANGED
|
@@ -12,7 +12,7 @@ from model import GPT
|
|
| 12 |
# Constants
|
| 13 |
# =====================================================
|
| 14 |
REPO_ID = "HayatoHongo/everyoneschat-checkpoints"
|
| 15 |
-
FILENAME = "
|
| 16 |
VISION_ENCODER = "openai/clip-vit-large-patch14"
|
| 17 |
NUM_IMAGE_PATCHES = 256
|
| 18 |
PAD_TOKEN_ID = 50256
|
|
|
|
| 12 |
# Constants
|
| 13 |
# =====================================================
|
| 14 |
REPO_ID = "HayatoHongo/everyoneschat-checkpoints"
|
| 15 |
+
FILENAME = "checkpoint_010000_vision_instructv2.pt"
|
| 16 |
VISION_ENCODER = "openai/clip-vit-large-patch14"
|
| 17 |
NUM_IMAGE_PATCHES = 256
|
| 18 |
PAD_TOKEN_ID = 50256
|