MCPcopy Create free account

hub / github.com/allenai/molmo2 / functions

Functions2,270 in github.com/allenai/molmo2

↓ 1 callersFunctionensure_finite_
Modify ``x`` in place to replace ``float("-inf")`` with the minimum value of the dtype when ``check_neg_inf`` is ``True`` and to replace ``fl
olmo/nn/ulysses.py:18
↓ 1 callersFunctioneval_caption
Evaluate caption using OpenAI API. :param prediction: predicted caption :param data: ground truth data :param openai_api_key: OpenAI
olmo/eval/video_caption_utils.py:227
↓ 1 callersFunctioneval_caption_consistency
( prediction: str, data: Dict, openai_api_key: str, )
olmo/eval/video_caption_utils.py:123
↓ 1 callersFunctioneval_caption_recall
( prediction: str, data: Dict, openai_api_key: str, )
olmo/eval/video_caption_utils.py:207
↓ 1 callersFunctioneval_captioning
( prediction: str, mc_question: str, mc_answer: str, openai_api_key: str, )
olmo/eval/temp_compass_utils.py:249
↓ 1 callersFunctioneval_multi_choice
Evaluate a multiple choice instance.
olmo/eval/mmmu_eval_utils.py:212
↓ 1 callersFunctioneval_multi_choice
( question: str, prediction: str, answer: str, openai_api_key: str, use_api: bool = True,
olmo/eval/temp_compass_utils.py:148
↓ 1 callersMethodeval_multi_choice
(self, gold_i, pred_i)
olmo/eval/evaluators.py:1824
↓ 1 callersMethodeval_multi_choice
(self, gold_i, pred_i)
olmo/eval/evaluators.py:1898
↓ 1 callersMethodeval_multi_choice
(self, gold_i, pred_i)
olmo/eval/evaluators.py:2009
↓ 1 callersFunctioneval_open
Evaluate an open question instance
olmo/eval/mmmu_eval_utils.py:229
↓ 1 callersFunctioneval_question_against_gt
(_args)
olmo/eval/open_ended_qa_eval.py:129
↓ 1 callersFunctioneval_question_against_image
(_args)
olmo/eval/open_ended_qa_eval.py:74
↓ 1 callersMethodeval_v1
(self, item_list)
olmo/eval/evaluators.py:2065
↓ 1 callersMethodeval_v2
(self, item_list)
olmo/eval/evaluators.py:2132
↓ 1 callersFunctioneval_yes_no
( question: str, prediction: str, answer: str, openai_api_key: str, use_api: bool = True,
olmo/eval/temp_compass_utils.py:183
↓ 1 callersFunctionevaluate_frame_predictions
Evaluate predictions for a single frame. Args: pred_points: List of predicted (x, y) coordinates gt_points: List of grou
olmo/eval/object_tracking_utils.py:561
↓ 1 callersFunctionevaluate_video_tracks_with_masks
Evaluate predictions for a single video. Args: pred_tracks: Predicted trajectory data Format: [{
olmo/eval/object_tracking_utils.py:650
↓ 1 callersFunctionexample_to_html_dict
Build HTML visualizations for an examples ex: The example (after preprocessing) to show preprocessor: The preprocessor used to preprocessor t
olmo/html_utils.py:222
↓ 1 callersMethodextract_characters_regex
(self, s)
olmo/eval/evaluators.py:1519
↓ 1 callersMethodextract_characters_regex
(self, s)
olmo/eval/evaluators.py:1627
↓ 1 callersFunctionextract_clip
Extract a clip from source_path[start:end] -> output_path using ffmpeg. Returns (success, message).
scripts/extract_coin_clips.py:52
↓ 1 callersFunctionextract_clip
Extract a clip from a video using ffmpeg. Args: video_path (str): Path to the input video start_time (float): Start time
scripts/extract_ego4d_clips.py:45
↓ 1 callersFunctionextract_events
(caption, openai_api_key)
olmo/eval/dream_caption_utils.py:151
↓ 1 callersFunctionextract_frames
Extract all frames from a video to JPEG files using ffmpeg.
olmo/data/academic_video_track_datasets.py:192
↓ 1 callersFunctionextract_image_points
Extract points from MolmoPoint image output text return points: [n_points, 4] array of (object_id, image_num, x, y) points
olmo/hf_model/modeling_molmo_point.py:942
↓ 1 callersFunctionextract_numbers
Exact all forms of numbers from a string with regex.
olmo/eval/mmmu_eval_utils.py:136
↓ 1 callersMethodextract_options
(self, option_string: str)
olmo/data/academic_multi_image_datasets.py:173
↓ 1 callersFunctionextract_pred
Extract the yes/no predction from the original video llm output
olmo/eval/temp_compass_utils.py:172
↓ 1 callersFunctionextract_single_point_track_per_frame
Parse single_point_track_per_frame model output. Expected format matches data_formatter.py: "[x, y, time], [x, y, time], [x, y, time
olmo/eval/object_tracking_utils.py:254
↓ 1 callersFunctionextract_video_point_track_per_frame
Extract points from video_point_track_per_frame prompt style. Text format example: time 0.50 {0: [34.0, 63.0], 1: [72.0, 49.0
olmo/eval/object_tracking_utils.py:140
↓ 1 callersFunctionextract_video_point_track_per_frame_with_occlusion
Extract points from video_point_track_all_frames_with_occlusion format. Text format example (yes if occluded): time 0.50 {0:
olmo/eval/point_tracking_utils.py:67
↓ 1 callersFunctionextract_video_points
Extract points from MolmoPoint video output text return points: [n_points, 4] array of (object_id, timestamp, x, y) points
olmo/hf_model/modeling_molmo_point.py:975
↓ 1 callersFunctionextract_video_points_start_end
Extract points from video_point_ground_start_end format prompt style. Text format example: 0: ([34.0, 63.0, 0.50], [35.0, 64.0, 1.00]
olmo/eval/object_tracking_utils.py:196
↓ 1 callersFunctionf1_score
Compute F1 score from precision and recall. (Copied from olmo.eval.evaluators.py)
olmo/eval/object_tracking_utils.py:552
↓ 1 callersMethodfetch_videos
Convert a single or a list of urls into the corresponding `np.array` objects. If a single url is passed, the return value will be a
olmo/hf_model/video_processing_molmo2.py:732
↓ 1 callersFunctionfilter_warnings
()
olmo/util.py:191
↓ 1 callersFunctionfind_first_mask
(mask_list)
olmo/eval/object_tracking_utils.py:613
↓ 1 callersFunctionfind_first_padding_frame
Find the first padding frame for each batch. Args: pooled_patches_idx (torch.Tensor): Tensor of shape (B, NP, G) indicating the pooled pa
olmo/nn/temporal_image_vit.py:724
↓ 1 callersFunctionfind_source_video
Find the source video file, trying known extensions.
scripts/extract_coin_clips.py:37
↓ 1 callersMethodfind_video_path
(self, video_filename)
olmo/data/academic_video_datasets.py:282
↓ 1 callersMethodfit
(self)
olmo/train/trainer.py:1141
↓ 1 callersFunctionflatten_dict
(d: Dict)
olmo/safetensors_util.py:36
↓ 1 callersFunctionflatten_mesh
Flatten a multi-dimensional ``DeviceMesh`` into a 1D ``DeviceMesh``. :param device_mesh: The multi-dimensional ``DeviceMesh`` to flatten.
olmo/dist_util.py:237
↓ 1 callersMethodformat_annotated_text
(self, answer, point_annotations)
olmo/preprocessing/data_formatter.py:789
↓ 1 callersMethodformat_messages
Applies system formatting to ith message from a sequence of messages
olmo/preprocessing/data_formatter.py:1511
↓ 1 callersMethodformat_messages
Applies system formatting to ith message from a sequence of messages
olmo/models/molmo_point/molmo_point_data_formatter.py:705
↓ 1 callersMethodformat_multi_image_points
Format multi-image points
olmo/preprocessing/point_formatter.py:47
↓ 1 callersMethodformat_multi_image_points
Format video points for counting objects across frames.
olmo/models/molmo_point/molmo_point_data_formatter.py:285
↓ 1 callersMethodformat_multi_points
Format multi-image pointing and counting. Generates questions and answers for pointing to and counting objects across multiple images.
olmo/preprocessing/data_formatter.py:899
↓ 1 callersMethodformat_options_and_answer
(choices)
olmo/data/academic_video_datasets.py:4193
↓ 1 callersMethodformat_options_and_answer
(answer_idx, options)
olmo/data/academic_video_datasets.py:4292
↓ 1 callersMethodformat_options_and_answer
(answer, choices)
olmo/data/academic_video_datasets.py:4803
↓ 1 callersFunctionformat_timedelta
(td: timedelta)
olmo/util.py:631
↓ 1 callersFunctionformat_value
(value: float)
olmo/util.py:500
↓ 1 callersMethodformat_video_input_points
Legacy method for formatting initial input points for user prompt.
olmo/preprocessing/data_formatter.py:1260
↓ 1 callersMethodformat_video_object_track_points
Format video points for tracking objects across frames. Keep only frames that match actual video timestamps. Sample frames at
olmo/preprocessing/data_formatter.py:1386
↓ 1 callersMethodformat_video_object_track_points
Format video points for tracking objects across frames. Keep only frames that match actual video timestamps. Sample frames at
olmo/models/molmo_point/molmo_point_data_formatter.py:540
↓ 1 callersMethodformat_video_tracks
(self, point_tracks, scale, label, alt_text=None, rng=None, start_end_only=False,
olmo/preprocessing/point_formatter.py:865
↓ 1 callersMethodforward_no_pruning
(self, x)
olmo/nn/temporal_image_vit.py:170
↓ 1 callersFunctionfuzzy_matching
(pred: str)
olmo/eval/vsi_bench_utils.py:5
↓ 1 callersMethodgenerate_multipoint_query
Format multi-image pointing and counting. Generates questions and answers for pointing to and counting objects across multiple images.
olmo/models/molmo_point/molmo_point_data_formatter.py:305
↓ 1 callersMethodget
(self, idx, epoch=0)
olmo/data/dataset.py:68
↓ 1 callersMethodget
(self, item, rng)
olmo/data/dataset.py:104
↓ 1 callersMethodget
(self, idx, rng)
olmo/data/academic_video_track_datasets.py:509
↓ 1 callersMethodget_abc_image
(self, image, annotations)
olmo/data/dataset_builders/ai2d.py:118
↓ 1 callersFunctionget_all_keys
(dicts)
olmo/util.py:483
↓ 1 callersFunctionget_batch
()
tests/data/test_data_iterator.py:240
↓ 1 callersMethodget_buffered_example_ids
(self)
olmo/data/dynamic_packer.py:424
↓ 1 callersFunctionget_candidate_sampling_fps
(video_fps, sampling_fps, max_fps=MAX_VIDEO_FPS)
olmo/data/academic_video_track_datasets.py:211
↓ 1 callersFunctionget_canonical_statements
(model_caption: str, openai_api_key: str)
olmo/eval/video_caption_utils.py:88
↓ 1 callersFunctionget_captioning_eval_result
(prompt: str, mc_answer: str, openai_api_key: str, maxtry: int=10)
olmo/eval/temp_compass_utils.py:139
↓ 1 callersFunctionget_causal_attention_bias
(cache: BufferCache, seq_len: int, device: torch.device)
olmo/nn/llm.py:1548
↓ 1 callersFunctionget_chat_response
Run a query through an OpenAI model
olmo/eval/vixmo_caption_utils.py:15
↓ 1 callersFunctionget_chat_response
Run a query through an OpenAI model
olmo/eval/video_caption_utils.py:9
↓ 1 callersMethodget_chat_response
( self, prompt: str, sys_prompt: str = "You are a helpful assistant.",
olmo/eval/evaluators.py:1653
↓ 1 callersMethodget_clip_subtitles
Extract subtitles that overlap with the given time range.
olmo/data/academic_video_datasets.py:5740
↓ 1 callersMethodget_connector_parameters
(self)
olmo/models/molmo_point/molmo_point.py:526
↓ 1 callersFunctionget_consistency_statements
( gt_caption: str, statements_str: str, openai_api_key: str )
olmo/eval/video_caption_utils.py:103
↓ 1 callersMethodget_cp_buffers
(self, max_seq_len: int, device: torch.device)
olmo/nn/llm.py:1320
↓ 1 callersFunctionget_dataset_registry
()
olmo/data/academic_video_track_datasets.py:2077
↓ 1 callersMethodget_decoder
(self)
olmo/hf_model/modeling_molmo_point.py:1069
↓ 1 callersMethodget_decoder
(self)
olmo/hf_model/modeling_molmo2.py:1141
↓ 1 callersFunctionget_default_device
Get the default device.
olmo/dist_util.py:377
↓ 1 callersMethodget_default_model_name
(cls)
olmo/models/model_config.py:85
↓ 1 callersFunctionget_device_info
()
olmo/dist_util.py:48
↓ 1 callersFunctionget_dp_mesh
Get the data parallel sub-mesh associated from a ``DeviceMesh`` created by :func:`build_world_mesh()`. .. important:: This is the me
olmo/dist_util.py:336
↓ 1 callersFunctionget_dp_model_mesh
Get the right sub-mesh for a data parallel model wrapper like FSDP or DDP from a ``DeviceMesh`` created by :func:`build_world_mesh()`. .
olmo/dist_util.py:148
↓ 1 callersMethodget_file_name
(step, process_index)
olmo/eval/evaluators.py:277
↓ 1 callersFunctionget_file_size
Get the size of a local or remote file in bytes. :param path: Path/URL to the file.
olmo/io.py:87
↓ 1 callersFunctionget_float_dtype_by_name
(dtype)
olmo/eval/model_evaluator.py:56
↓ 1 callersFunctionget_font_size
Get the font size based on the size of the image.
olmo/data/dataset_builders/ai2d.py:22
↓ 1 callersFunctionget_frame_coordinates_in_collage
Transform coordinates from original image to resized image with padding. Args: x_orig: X coordinate in original image y_
olmo/html_utils.py:66
↓ 1 callersMethodget_frame_selection_parameters
(self)
olmo/models/model.py:143
↓ 1 callersFunctionget_frame_times_and_chosen_fps
(selected_sampling_fps, total_frames, max_frames, video_fps)
olmo/data/video_loader.py:96
↓ 1 callersFunctionget_frame_times_and_chosen_fps
( selected_target_fps, total_frames, max_frames, video_fps )
olmo/hf_model/video_processing_molmo2.py:534
↓ 1 callersMethodget_fsd_args
(self, autocast_precision)
olmo/train/trainer_config.py:133
↓ 1 callersMethodget_fsdp_wrap_policy
Get a FSDP1 wrap policy for this model.
olmo/models/model.py:130
↓ 1 callersMethodget_gcs_blob
(resource: str)
olmo/io.py:987
↓ 1 callersFunctionget_gcs_url
(output_file)
olmo/eval/model_evaluator.py:75
← previousnext →701–800 of 2,270, ranked by callers