Update README.md
Browse files
README.md
CHANGED
|
@@ -85,7 +85,7 @@ from PaDT import PaDTForConditionalGeneration, VisonTextProcessingClass, parseVR
|
|
| 85 |
|
| 86 |
|
| 87 |
TEST_IMG_PATH="./eval/imgs/000000368335.jpg"
|
| 88 |
-
MODEL_PATH="PaDT-MLLM/
|
| 89 |
|
| 90 |
# load model
|
| 91 |
model = PaDTForConditionalGeneration.from_pretrained(MODEL_PATH, torch_dtype=torch.bfloat16, device_map={"": 0})
|
|
|
|
| 85 |
|
| 86 |
|
| 87 |
TEST_IMG_PATH="./eval/imgs/000000368335.jpg"
|
| 88 |
+
MODEL_PATH="PaDT-MLLM/PaDT_RIC_3B"
|
| 89 |
|
| 90 |
# load model
|
| 91 |
model = PaDTForConditionalGeneration.from_pretrained(MODEL_PATH, torch_dtype=torch.bfloat16, device_map={"": 0})
|