Update frontPrompt.py
Browse files- frontPrompt.py +0 -1
frontPrompt.py
CHANGED
|
@@ -27,7 +27,6 @@ def load_image(image_file):
|
|
| 27 |
|
| 28 |
|
| 29 |
def main(image_path,model,tokenizer):
|
| 30 |
-
path = "OpenGVLab/InternVL2_5-4B"
|
| 31 |
pixel_values = load_image(image_path).to(torch.float32).to("cpu")
|
| 32 |
generation_config = dict(max_new_tokens=1024, do_sample=True)
|
| 33 |
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
def main(image_path,model,tokenizer):
|
|
|
|
| 30 |
pixel_values = load_image(image_path).to(torch.float32).to("cpu")
|
| 31 |
generation_config = dict(max_new_tokens=1024, do_sample=True)
|
| 32 |
|