Update vision_analyzer.py
Browse files- vision_analyzer.py +5 -4
vision_analyzer.py
CHANGED
|
@@ -378,9 +378,9 @@ def analyze_single_frame(image_path: str, preprompt: str = "") -> dict:
|
|
| 378 |
input_ids=inputs["input_ids"],
|
| 379 |
attention_mask=inputs["attention_mask"],
|
| 380 |
pixel_values=pixel_values,
|
| 381 |
-
max_new_tokens=
|
| 382 |
-
num_beams=
|
| 383 |
-
early_stopping=
|
| 384 |
pad_token_id=vision_processor.tokenizer.eos_token_id
|
| 385 |
)
|
| 386 |
|
|
@@ -763,4 +763,5 @@ __all__ = [
|
|
| 763 |
"extract_frames",
|
| 764 |
"DEFAULT_FPS",
|
| 765 |
"ensure_dir"
|
| 766 |
-
]
|
|
|
|
|
|
| 378 |
input_ids=inputs["input_ids"],
|
| 379 |
attention_mask=inputs["attention_mask"],
|
| 380 |
pixel_values=pixel_values,
|
| 381 |
+
max_new_tokens=500,
|
| 382 |
+
num_beams=5,
|
| 383 |
+
early_stopping=False,
|
| 384 |
pad_token_id=vision_processor.tokenizer.eos_token_id
|
| 385 |
)
|
| 386 |
|
|
|
|
| 763 |
"extract_frames",
|
| 764 |
"DEFAULT_FPS",
|
| 765 |
"ensure_dir"
|
| 766 |
+
]
|
| 767 |
+
|