Fred808 commited on
Commit
899695c
·
verified ·
1 Parent(s): e1cd712

Update vision_analyzer.py

Browse files
Files changed (1) hide show
  1. 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=50,
382
- num_beams=3,
383
- early_stopping=True,
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
+