oumi
View on PyPI — Reverse Dependencies (1)
0.1.8 | oumi-0.1.8-py3-none-any.whl |
Wheel Details
Project: | oumi |
Version: | 0.1.8 |
Filename: | oumi-0.1.8-py3-none-any.whl |
Download: | [link] |
Size: | 513818 |
MD5: | 1a5c9c8133ab93c97f1f1f44f347bb67 |
SHA256: | 0d742079d7c2f9ba22e14b46b4e1204cf4cd5f4ec1a436e8508675b2d73f9f26 |
Uploaded: | 2025-03-10 18:29:13 +0000 |
dist-info
METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt
METADATA
WHEEL
Wheel-Version: | 1.0 |
Generator: | setuptools (76.0.0) |
Root-Is-Purelib: | true |
Tag: | py3-none-any |
RECORD
Path | Digest | Size |
---|---|---|
experimental/__init__.py | sha256=eke0C_t-RULyOs0auL6b3yxFpwMDE4tcoHyR__8efec | 660 |
experimental/configs/projects/zephyr/README.md | sha256=H4nBKmVDCOQVjZt71A2UBsJNkmVC_hm3vNcyBdgp8lU | 254 |
experimental/configs/projects/zephyr/evaluation/eval.yaml | sha256=O77b44DIZOKbXWy5ggYl_bUCOIwQFiS4n6oVhIXBONU | 1081 |
experimental/configs/projects/zephyr/sft/full_gcp_job.yaml | sha256=GjTyA6adLxXPFalxEEeT7x704FinGLpRGl46NW-pWHE | 2055 |
experimental/configs/projects/zephyr/sft/full_train.yaml | sha256=ogLeQ8q7dpEu37riQ_MrnCXrPyNag_Ih4AoD7PEXQMs | 1989 |
experimental/configs/projects/zephyr/sft/qlora_gcp_job.yaml | sha256=LmUokhTf7atm1oWboXC9uDnwuw6SxucCSQqgbiHLRRI | 1983 |
experimental/configs/projects/zephyr/sft/qlora_train.yaml | sha256=pvb-6cJu099AEsm-f68zyHAjTBkA0jZw1cj7zILDg3g | 2296 |
experimental/configs/recipes/phi3/dpo/fsdp_gcp_job.yaml | sha256=4-pVMR2xWl-Eh7BkWrJ9vhJGGVDOQrK6pTG_pqAu6uQ | 1400 |
experimental/configs/recipes/phi3/dpo/fsdp_nvidia_24g_train.yaml | sha256=1MKTe1YLrEG0a9NnisWDulKcgeNYUZpiY3KS36OlaJY | 1404 |
experimental/notebooks/Oumi - Datasets Tutorial.ipynb | sha256=M7lOAee5S2b8EU7cMR7zOFtB3zZh5ki936muiurhdJU | 18629 |
oumi/__init__.py | sha256=XY3aN7-F_XRxoNWNzxI6UOBShrSAi47Scqt1CVRxNOk | 8381 |
oumi/__main__.py | sha256=42d6kX8b6OKA3GSCXOpmYVivBNvaGUDXPvl8tfmayDk | 1189 |
oumi/evaluate.py | sha256=MAr4V6XTJtNFSjSAOgWOVDTEnN4W5xfpF3pHY9K5Nrk | 1279 |
oumi/evaluate_async.py | sha256=s7XQMIwKkev8vu3VwgEdZ1AMv4Y30MAhAVB7ueNj8CE | 4291 |
oumi/infer.py | sha256=MAXv3mZNb0lDf5q4PwVlEA2KbKblpKJqUeru0Sfc8kg | 4950 |
oumi/judge.py | sha256=4_2TSHmLrNMkTTvq6wnXtnKUbE6SellGxxKg_CzUD34 | 3323 |
oumi/train.py | sha256=5FcI-vEmi-1fjWpZ5qL0W3KJlwLk9b3AwCc3zo2TGbg | 15042 |
oumi/builders/__init__.py | sha256=-K-6ho4vRZ2rsScxflyRJ3epEu_QWR_idzhZ5hG7UC0 | 2084 |
oumi/builders/callbacks.py | sha256=CIGbq5bKp9VDpqZBA6AKL6PfNYDl86qAOsdJRD0EZHY | 5526 |
oumi/builders/collators.py | sha256=kqu6r7nALoXm1nUh51zACtWOOi3RAoqjeJs7SrPJ4XQ | 6940 |
oumi/builders/data.py | sha256=UV5LF6x9RCxkqAC_e28aGIz7lM_Su6NjebREJ0WtciQ | 12614 |
oumi/builders/inference_engines.py | sha256=K9T8-O5ZWIc_VMyW4R8r8BEsaWNKb7wvfBcMWv1PVb0 | 3590 |
oumi/builders/lr_schedules.py | sha256=yjcrcsHxGj7iZpYO4NYx8BFl7QZGvxn9jytr9XCBqHc | 6794 |
oumi/builders/metrics.py | sha256=fs7hg50eA4zRYhDHrwebTfCyGoCos2eoxtVqTQSI9yQ | 1170 |
oumi/builders/models.py | sha256=mKRAbFvQ2K_5upRlOJcOHcTVi_Gcjj4NJe2_zAAa5VU | 21893 |
oumi/builders/optimizers.py | sha256=BCr55pBMxHlSs1FEPgXHAsKVM7CPXq7ioPn0SGlkXiY | 3919 |
oumi/builders/oumi_data.py | sha256=-s8lOBkF9iUKPpN_FpZrzGBca_-hyOLmPP7QkQb9bPY | 7134 |
oumi/builders/processors.py | sha256=kwKXTGJ9N4cc2rCRPDxZz5iKJhOF2ta4DY45UqXMTic | 2589 |
oumi/builders/rewards.py | sha256=gu77gC72n78qtdMHIirHbigi-nVW4XCQxIfwPASdvzo | 1421 |
oumi/builders/training.py | sha256=cjXRoVHS1GSbtChIDaj8MDkT5a_HfuLkZqRNQPGqW9Q | 5077 |
oumi/cli/cli_utils.py | sha256=LqwsjaLLkdJJw0H2vIq8yX434yO-GvZMFQRZzODcyrk | 7190 |
oumi/cli/distributed_run.py | sha256=usonMR11RtOJJdEEHFVV_iznSxO6e2HVq1xX-601zrk | 14902 |
oumi/cli/env.py | sha256=yxcq3cBDPRWiG6CUTCdbQYbrdz3i1gIOnZp9oE6TySI | 5059 |
oumi/cli/evaluate.py | sha256=8E1CxlY4Nrxo8t3e06DtHaJbhnE831QhtUHY4mtgQX8 | 1691 |
oumi/cli/fetch.py | sha256=u97gYEKxnen0eHWK15SQ1wgt3TyuqqYYtxSRwz8YEXY | 1761 |
oumi/cli/infer.py | sha256=cGtf7WXce_LyqPdWWcub8kXntYMEEvy-W-DcQO7pWOg | 5091 |
oumi/cli/judge.py | sha256=9WXVF6hgIL36AEktWGK1Gtz6Hs793p3kx7n3V291s20 | 7651 |
oumi/cli/launch.py | sha256=ih_4jUz_fgNwvDh-49JfNZ1LW-WjrR9OYwXzAoRDFg0 | 17091 |
oumi/cli/main.py | sha256=XP4xa8CRD1-TAZF8pvmHZ4fW6jWgCFSQJBlRrhNaMHk | 4154 |
oumi/cli/train.py | sha256=G-UQmhC4HJy4-n731xweFgek7TIQVQQS0MhJPa2og5A | 2025 |
oumi/core/__init__.py | sha256=i1h7WkVWLgMqpmKOrUSMDQ8wTNhOsnADShp0fOzANu8 | 804 |
oumi/core/async_utils.py | sha256=IouIJaEU41OjQLuo5_foFafVXZwyHRZ4f11CM4AjRzs | 1383 |
oumi/core/constants.py | sha256=CW8P_3sI1NR4bjZWCBJ57aY_Tm6ULl5SaGWLk-sFpZQ | 933 |
oumi/core/distributed.py | sha256=4CuyGumRnnMYRAEhMKkbGcSZLFkRADkTwbFYNjF31bI | 20685 |
oumi/core/callbacks/__init__.py | sha256=0bl1f22gALKBkoCuWW3AACTDsk-5Jj7IiR1aQxniSwk | 1500 |
oumi/core/callbacks/base_trainer_callback.py | sha256=ekPset4rYHvbQnPkn2-mTOhDJwZTk1Cre7upQPRs8PY | 643 |
oumi/core/callbacks/hf_mfu_callback.py | sha256=YIyAhVaIx1bHpjX9yTp9EhmyzVqe31fs_FF9jSfklds | 6112 |
oumi/core/callbacks/mfu_callback.py | sha256=Rntbfgeawr_14dS3iqqZBSMVgYFMtpbq_YsUALfFfto | 7740 |
oumi/core/callbacks/nan_inf_detection_callback.py | sha256=voIqEP9WCLRBFRSHtCDju2xSlm3SCBNGTFw8MeMi3Bg | 2358 |
oumi/core/callbacks/profiler_step_callback.py | sha256=sWT9FTkgUSKr1snqGnWblOHqOW3cLU0bqnA5RkQhYrU | 3077 |
oumi/core/callbacks/telemetry_callback.py | sha256=mSxDrHqkje0HaqFuLuSOycSgi_zAPRXaiB3_QaBkLS8 | 11672 |
oumi/core/collators/text_collator_with_padding.py | sha256=Hrxb-KV9ypsL9V_PdC7lhDLrXCKQ_krZi8qVaOhlAAs | 10096 |
oumi/core/collators/text_completions_collator_with_padding.py | sha256=K5zcgvnpslPgqI1JPhyUbxwe6fJABaD_myRk552QpOw | 2285 |
oumi/core/collators/vision_language_collator_with_padding.py | sha256=jxU3ay0gFDtgwiLA5CpdAHIKP89EKuKA4RqOrLzl7hw | 5649 |
oumi/core/collators/vision_language_sft_collator.py | sha256=Gk-dj-DbT1yYvRX-_RTLH2O7hwMOidv8PdAos0phjuc | 3949 |
oumi/core/configs/__init__.py | sha256=c2ReKsJydSbwd-3won2D92ek-slVU0cRRqWSHVYn634 | 6344 |
oumi/core/configs/async_evaluation_config.py | sha256=F-2UYWjqkFcq4cR99T3IGeJ_HUGTWTJ5BrRVXhGw9SA | 1996 |
oumi/core/configs/base_config.py | sha256=6JNbVbH-Ny22t3UlPxUx6iT7jU5t8StkhHzsAF-UbY8 | 6724 |
oumi/core/configs/evaluation_config.py | sha256=ymW6eoJzpa2iZJR-STx6R99Wkb_39MspvMee4MWEioo | 3178 |
oumi/core/configs/inference_config.py | sha256=fvHvavLjEBRm-UXnZkSv3su1yHk09U1oEeLKQ0Xcl2A | 2399 |
oumi/core/configs/inference_engine_type.py | sha256=7QVhhHCeLfuAIVS6zRBpdMIFKXoTeFrADyWwfNxPzLI | 1812 |
oumi/core/configs/job_config.py | sha256=vmBLZAdKROJAW3OUCmal7kZ_9gN1A2ge_EMFZtOCTwo | 5030 |
oumi/core/configs/judge_config.py | sha256=G34e653nDi6wz186uIMH82ySnQ-97g3oURbdHxNVpKY | 5916 |
oumi/core/configs/training_config.py | sha256=3WMtjoP0ppOzpxN6rfn9wnuvpf_yf4Brr0RPTjzDnlk | 5109 |
oumi/core/configs/internal/internal_model_config.py | sha256=PD_P3-o1WRnKmbtLK16Tr3VBkG_9OybPADuldhhryRM | 4566 |
oumi/core/configs/internal/supported_models.py | sha256=ENvRoNZuYqLHlJmRlwAIHJwSyc1nBMbvyxvDrqlI2rI | 11694 |
oumi/core/configs/params/base_params.py | sha256=3lG9UAgicNoli4S_T3_IVOHKRwxVxZmv9vsKTO473qk | 3174 |
oumi/core/configs/params/data_params.py | sha256=IIVJ6tfPyH57HC0UkDU5I5NBWUQzOoMRR-yqbvTGol0 | 13200 |
oumi/core/configs/params/evaluation_params.py | sha256=1jNBzoDoQbtkwqx3mhZ92GWiXABYG-h1tvwlSMVb5gg | 8474 |
oumi/core/configs/params/fsdp_params.py | sha256=KoC9lo4ehXgmLc8rDLzeXf_vOdFgiZCKkBVgwKitpac | 8899 |
oumi/core/configs/params/generation_params.py | sha256=FIL0mVn0-dVpn8-9pu8g8dCj5slf4aUJDneL3_14eF8 | 5673 |
oumi/core/configs/params/grpo_params.py | sha256=9f_lcmQtBt6E80NrQ7YyP2BVvC8t7tCDQRw2alfqpMg | 6306 |
oumi/core/configs/params/guided_decoding_params.py | sha256=NEwJdPSGOqHdESiJdHq6gdNyNJjHBRsbQIP7isnNTM0 | 1997 |
oumi/core/configs/params/model_params.py | sha256=c7_xNcflm9UNJvclaRTlgo51sDo88EVWIJfw5CjMRFo | 9933 |
oumi/core/configs/params/peft_params.py | sha256=jb7Prak50x776OWwiPNyYVIUm6E1tnD0NTZKG0h9oBc | 10384 |
oumi/core/configs/params/profiler_params.py | sha256=3x4pox3ZztJjL7_z9bpZUIcgDiYFGSoSwp64j86mMpg | 4504 |
oumi/core/configs/params/remote_params.py | sha256=CKhMEkLmKOj0PFSN-_rQOx8wRFtVbJAUbrlKR3NJ8-Q | 2505 |
oumi/core/configs/params/telemetry_params.py | sha256=T6JocFiSuPJ-GRrjGBOqosFIvHIKsNszFxF2qiAsHbY | 1486 |
oumi/core/configs/params/training_params.py | sha256=AHWomRfh9pwr6TMLRjTKNDBEY0eyEWslxrjYD9QbE4M | 30720 |
oumi/core/datasets/__init__.py | sha256=FhbK7IRVs_GGWqL1_ApTqkW1UUEEKotahSgk5-r0LUg | 2039 |
oumi/core/datasets/base_dpo_dataset.py | sha256=aJEuX13mjRvQs9nlrgZFgG4HfPRUHQIbNQXrZmv7Qsc | 2934 |
oumi/core/datasets/base_grpo_dataset.py | sha256=85M7iQM_Wbvyql3zvARVfyAYm13ABjb8wWOoVm6yxbM | 3313 |
oumi/core/datasets/base_iterable_dataset.py | sha256=1UdxsteUI6Xk2HTiel6g5gxgGLF5s-KI3YQd-3BBTEs | 4257 |
oumi/core/datasets/base_map_dataset.py | sha256=l8F-5kq7Q9hZ09zQLmAB6RwxYJPsb3EIwoycrRyI0Tk | 18697 |
oumi/core/datasets/base_pretraining_dataset.py | sha256=CLA1Rxcxtf26vPBwINx2ghm4NGjC5xLLyXXuL5ISDH0 | 5437 |
oumi/core/datasets/base_sft_dataset.py | sha256=OAo2aoF1NbmaoOahC-OuZ3uLbZown1lrpMYJ4xzWjPc | 10283 |
oumi/core/datasets/packed_sft_dataset.py | sha256=KP2OHtpq5Artc97VuebEBilxZ73AH7YHpRJoaO-dgHQ | 11312 |
oumi/core/datasets/pretraining_async_text_dataset.py | sha256=Z0R2fzI0lW37m0f_HAbwtoX-TBBJZW6W-_ikBafnPx8 | 9981 |
oumi/core/datasets/vision_language_dataset.py | sha256=WIgrTmBjD5ZN_LQaQh_lkVIM_Scl7C_XXC4KtlR3Pm4 | 5749 |
oumi/core/evaluation/__init__.py | sha256=FxQhBNajmuyw1htU7RV0luv2mUQ6tHxZKioZ7k5P1f4 | 1438 |
oumi/core/evaluation/evaluation_result.py | sha256=omARay6Y9mV4_IuEFD7zxtgbP_h0a6EIP4fW8-SiaEc | 2453 |
oumi/core/evaluation/evaluator.py | sha256=NVVAC416ZWFSLCHzFMuhRwaXdjcAYgiv92Ew57l_WWY | 14273 |
oumi/core/evaluation/backends/alpaca_eval.py | sha256=eK7adKFzgVYHEwHILvXlZz_QhQd6y7EGR7X4oA0NUIY | 6372 |
oumi/core/evaluation/backends/lm_harness.py | sha256=gtYgXlYz3rgi-svZci8GeR5p8zdgZ_LYnJI6evz0klE | 19610 |
oumi/core/evaluation/utils/platform_prerequisites.py | sha256=tIgi3Hn4NKRGzuD5KKph8E9g-sP7bQ7Zpddb_KgzpPI | 3770 |
oumi/core/evaluation/utils/save_utils.py | sha256=H4ydsRrd59UJAQdmaBQRzy6OAkWr0ElwX6rtb2yBT4Q | 6547 |
oumi/core/feature_generators/__init__.py | sha256=ZllL9m3SYJYyBQ_c-RaOS6vwn470sqCE5UYf9daLGGU | 1221 |
oumi/core/feature_generators/base_feature_generator.py | sha256=4Nb8EDwI5kspKcIByVXyI7kVWuSaX1anpln7i4RvPn4 | 2097 |
oumi/core/feature_generators/vision_language_conversation_feature_generator.py | sha256=hAQlN7kea3_8OOh1HGCoJYHoeSJl34dpFUiRs8M1yKE | 14549 |
oumi/core/inference/__init__.py | sha256=LxbPsGIz6w7um728wGYZKBJYcUL5qwOzz1mlEPz3mA0 | 850 |
oumi/core/inference/base_inference_engine.py | sha256=lhjZNmj9hQBoTWxdBfEMdFyvWT08SNstRlCuVvJcoSg | 8959 |
oumi/core/launcher/__init__.py | sha256=0Lb4WQt9u-31LAvDnmaXfYNF2_SAWhC8uuy6bAs3xVs | 1067 |
oumi/core/launcher/base_cloud.py | sha256=D_V0JrHQoqYrHqsTwNrPGb0AieEKvFOrvaS8cyTBhCI | 1405 |
oumi/core/launcher/base_cluster.py | sha256=yddwpKNq2MKMbqPsAIEBeU5E9CyCTZfzFr0dR2XZU_Y | 2327 |
oumi/core/models/__init__.py | sha256=K-NmIwNJ0WgYbvjQIMEA5YTConTUfc6FfeAg4EI3GLM | 1384 |
oumi/core/models/base_model.py | sha256=kW-GoQoCwUWb3BmYEoAZPE2dx-g9P7DOjogYQDcAZDs | 1920 |
oumi/core/processors/base_image_processor.py | sha256=CG5odtOSgi_KvIp1KvofJemc5Z36y7fxZZiNlCSXOTs | 1215 |
oumi/core/processors/base_processor.py | sha256=LmsGUS08eYmhaUgJ-nYYn1wnJsPbJN0Z3CWo_gVmDew | 4015 |
oumi/core/processors/default_image_processor.py | sha256=jdH74urasa3gdiLp2yLAzlvkOp1uzwzdzLb-E4aETC4 | 2153 |
oumi/core/processors/default_processor.py | sha256=wI9qjH4uW9Lik-z1SvpVB3H6L3wwtBroS4aSA53ScT8 | 9105 |
oumi/core/registry/__init__.py | sha256=_Z6KF-TSwRtKQfYkWXydSEwyPVfo_i91n6t_PSFt7O8 | 1049 |
oumi/core/registry/registry.py | sha256=icK06kEfEm3M7C8wMCGrlCMkKOfWtF4AXzndqqnvGMI | 13265 |
oumi/core/tokenizers/__init__.py | sha256=k_pnkRETGTs-Gjf3YSYZdGZ18Kkis20CpnCODO25TxA | 1060 |
oumi/core/tokenizers/base_tokenizer.py | sha256=B8U9eYhbMlMpb0H-opCXSKWE0zGj5sbd_expHr1ci5I | 645 |
oumi/core/tokenizers/special_tokens.py | sha256=o4PWpxtECFs-uGXndohhJMVkWq5jzJvgiH65H89ap0k | 3647 |
oumi/core/tokenizers/utils.py | sha256=wj22a8YftvOSRAt3rvTLv9FmsmH2FIrrv6-_aRiWJ80 | 5585 |
oumi/core/trainers/__init__.py | sha256=rnkoaj_jgb6FUVzj0fnzZC3ywGR4B-wIJnzAZVjsjfQ | 1385 |
oumi/core/trainers/base_trainer.py | sha256=Fe5C1l1ul29n6O12vPtMtk5CFtw0vjz0AbA_WCpWhDk | 1574 |
oumi/core/trainers/hf_trainer.py | sha256=iWcRdmN1L4-VJxOwEexKtnaJ5TKVHT5Ct-NLFERxOHE | 5986 |
oumi/core/trainers/oumi_trainer.py | sha256=1qx583YcF02irUVA_axnZquoEcHOpzaUKWo9xyNx36M | 31317 |
oumi/core/types/__init__.py | sha256=P3PP7rJrth2-QknvsZHgCOOTh3EAMXEPkFAO3F8ramM | 1599 |
oumi/core/types/conversation.py | sha256=8q34I3TvzqiBLWNZbTEXwZIOoCoABbAbcdT-ff71VXc | 16996 |
oumi/core/types/exceptions.py | sha256=C6adQJQwdbNi6CXJPj2r-yD3VF3Xu1BVnnnNtQd8qIg | 675 |
oumi/datasets/__init__.py | sha256=R90QPoZzOjChPyeRrhyvVJf5TSBO8kwRKf2mcZY5zaI | 4926 |
oumi/datasets/debug.py | sha256=tQuU02iP5MCBfPNArKFjwHuyTMACYDyaJewvG0HDrlI | 6158 |
oumi/datasets/mmlu.py | sha256=M_eJc-q5mw_u1kCRlfvsOls2LFLWTygOx2u-DJudfbw | 8440 |
oumi/datasets/chat_templates/chat_ml.jinja | sha256=x-UssV0zyIdRbp2sJk9qe_Jem2T43KsZ_roUdyMpeiM | 539 |
oumi/datasets/chat_templates/default.jinja | sha256=nr5W5wyVnKQhPp8Y3h9TNP0itsI7YTdNfoPoJhbJvnM | 321 |
oumi/datasets/chat_templates/default_gen.jinja | sha256=Mqq8aMLwwb5kNd1TFRpDmhLrHhM3sfH_0wOIDi0OWhM | 574 |
oumi/datasets/chat_templates/gpt2.jinja | sha256=BoI0YzGXW8LzImT0aUUpU0V9fTAf32GAHlbYRKaGwNk | 78 |
oumi/datasets/chat_templates/llama3-instruct.jinja | sha256=DKxmF_LGEzHsT1yGNbbAu-3TsEbhXV66sWw9CNar4h8 | 875 |
oumi/datasets/chat_templates/llava.jinja | sha256=XOwQDQS0bs-u-IkERnibM-kKXw_3oIevGg7ADhKhSe8 | 1255 |
oumi/datasets/chat_templates/phi3-instruct.jinja | sha256=R3MmbJOLhOaYWCrUm-zvrJbaonqAQO25cTBisUqh-TA | 836 |
oumi/datasets/chat_templates/qwen2-vl-instruct.jinja | sha256=cdYd1NpB9RPfWwowULSKPijgN70Mvc16dBcMWEwZvTQ | 1471 |
oumi/datasets/chat_templates/zephyr.jinja | sha256=nPz1G1pbjCmd5-BWql6xcJK3gwmvSfiAC6hDVu31XCI | 509 |
oumi/datasets/evaluation/__init__.py | sha256=AemHJU6unV_lbcFIiy7N8KKzwxhnCeRoPPdweVMoTu0 | 708 |
oumi/datasets/evaluation/alpaca.py | sha256=FiVL_7ghRUzyhaQtoXtGTHctah6FJkGV5_gclKMh7po | 4268 |
oumi/datasets/evaluation/utils.py | sha256=__3ZM26eVrrehmacvyuZak2vKajJxD9mjluw7w_8Yt4 | 3583 |
oumi/datasets/grpo/__init__.py | sha256=Il5GzTe9fM7-Ci89nMJCPhrSNGx1GJ8HWE6LHUIHf7o | 690 |
oumi/datasets/grpo/tldr.py | sha256=Ur0z4lUVcuUmwDBndknXnpmN0v1K4L_utdKhvWNEmfg | 878 |
oumi/datasets/grpo/rewards/__init__.py | sha256=iTX8KdXAaRcgSXvZpW2zoJjTQzKvl-OYebXZJ5Nsnv0 | 870 |
oumi/datasets/grpo/rewards/completion_length_rewards.py | sha256=x4DgXs517Hv93Qa_zydX0jDvHriJwVRm8K-cCPBmSS4 | 3825 |
oumi/datasets/preference_tuning/__init__.py | sha256=QvCVpliyB6xU6O5myzfCzVdPvw6NWfzxmuKA5_8P3_U | 734 |
oumi/datasets/preference_tuning/orpo_dpo_mix.py | sha256=-yj60rvI3V7c8NpdxYU4z3YhFobsDLg-qp2IMoV894Y | 1892 |
oumi/datasets/pretraining/__init__.py | sha256=_PY8Art0csISF0VUmpXxH1wOSUZRHFr0au2_v9ozRNs | 2006 |
oumi/datasets/pretraining/c4.py | sha256=N_dy-pR5xpoPx5DbLJvhOLxvOmLNnH0OkpExMUZROv0 | 1751 |
oumi/datasets/pretraining/dolma.py | sha256=EDG40qCST2raMPP3UTmdJoepvAXP2PGtIWg1dMWax_w | 2244 |
oumi/datasets/pretraining/falcon_refinedweb.py | sha256=x0AsHLKcKUcrV4AUPwZV0fBmSuiq2LTID1LUKEPallM | 2399 |
oumi/datasets/pretraining/fineweb_edu.py | sha256=J6O5Dh4rtc1pqdfYqugyKjeTZyA2YJeEBzRH14bsGYQ | 1849 |
oumi/datasets/pretraining/pile.py | sha256=fxjhDnVSgeSo3n3xJVbHBcid7aMbpt8TIaH0DDFi9H4 | 2190 |
oumi/datasets/pretraining/red_pajama_v1.py | sha256=VkC1QlAiI53fCoNL-tc_NKGQzDDNAork7qq31YiwVPQ | 2431 |
oumi/datasets/pretraining/red_pajama_v2.py | sha256=dIqm4OO7PGGidhjvP7JBeXVr1cFW_LwagSdcMBxY3xM | 2023 |
oumi/datasets/pretraining/slim_pajama.py | sha256=cuOT8jZvwhKSCqJbclpTUJJBGUMRhfNOLNHu_GKHQsw | 2364 |
oumi/datasets/pretraining/starcoder.py | sha256=bqD6fdpOrRMFa3KMH-bnowRz5Kb4ABqyW-xphZ_rauU | 2449 |
oumi/datasets/pretraining/the_stack.py | sha256=8tWBANEH2IIxajddpLKurM8kumXCMipU2DT8t9HlQb4 | 2943 |
oumi/datasets/pretraining/tiny_stories.py | sha256=_xn4kr0J8k9xR8nzPvAJwRO31z4hEaVjelnIr0ROvog | 1334 |
oumi/datasets/pretraining/tiny_textbooks.py | sha256=F8eXKieOls2iyOSDZJLhA79PQys9ie2ThR8pYfnBuDU | 1765 |
oumi/datasets/pretraining/wikipedia.py | sha256=jPIV0gZy7vKsR3wE14qDBGEujOA4GOcGll4ANPPSojQ | 1900 |
oumi/datasets/pretraining/wikitext.py | sha256=yJ7IdexnIisku1ljROM-N-bNueKEx7wPANNVcCtjkzw | 1771 |
oumi/datasets/pretraining/youtube_commons.py | sha256=qJdRmBg2oQ1lsqkF6_ieH-6QcV64hv4sX-gf4J4TKH8 | 2301 |
oumi/datasets/sft/__init__.py | sha256=A18ZNmY0jLPZMnQimIyczAfv2xnfAsrFSbtik7K7QRQ | 1734 |
oumi/datasets/sft/alpaca.py | sha256=0XLzUBj4xUyn7-jWPFvSrImh-YacIlZgJNZOLmyl62I | 3119 |
oumi/datasets/sft/aya.py | sha256=F8NusEUYPbus7DnLW--UFghxDD_wCsbnk5n2Bceygtg | 1442 |
oumi/datasets/sft/chatqa.py | sha256=bmSACdTpS3F3dUs3KOZ_ehzejvZAEHIXLQWsciAsZF4 | 6372 |
oumi/datasets/sft/chatrag_bench.py | sha256=VCFwypweMYii6RloKfATVsxXD8RNfLzZtlh5MUe_LyY | 6050 |
oumi/datasets/sft/dolly.py | sha256=gN339YBq_HbbiOLPFRkwhTvzVUHZSjkTIP9hNB2AS1I | 3382 |
oumi/datasets/sft/magpie.py | sha256=zRJyUkX95j3jOjCZ6AuVgAdXBmrIRVFKK2N0dmGYi00 | 2547 |
oumi/datasets/sft/openo1_sft.py | sha256=ar-yqZIN4YAavVjJoAgHlajLicU1SdHt2AGIo1gw0AA | 1240 |
oumi/datasets/sft/prompt_response.py | sha256=zEFL2zifdCjdSiuwNTDMk4TsBP3iw3xh5cyosqK16P4 | 2580 |
oumi/datasets/sft/sft_jsonlines.py | sha256=6OAA9gQnLFEmz2vfpExXS4cLYA3tEXKN0dxURjbvcHg | 9375 |
oumi/datasets/sft/tulu3_sft_mixture.py | sha256=hkYaMZYLJTONemH0HdJ3RC794OTbOriZGNKQrAApSxo | 1946 |
oumi/datasets/sft/ultrachat.py | sha256=y3pjWPUqwg6s6WMpFL9ghf5vJHcbzJ_rUSGA96T_9Ss | 1409 |
oumi/datasets/sft/wildchat.py | sha256=yMAYCF1-a7LZCJXIajpO-NhSj-P3kdiwnrQIGtv_DcI | 1405 |
oumi/datasets/vision_language/__init__.py | sha256=bOUGfU03e21tHRtDEsQnn0bJLMSkrpwI1NpjauJg-4o | 1442 |
oumi/datasets/vision_language/coco_captions.py | sha256=tMrQiENq4B07xGUtqG7iEYjzdEtSvj7Bkkw50c3N8Ak | 3218 |
oumi/datasets/vision_language/docmatix.py | sha256=WQ0neKdvP0FDENfzj_IZD9gTpXMlL5F76pgGDuzwtIE | 1538 |
oumi/datasets/vision_language/flickr30k.py | sha256=Zb4nARzGE85D6pLpOf6QQim_9e99jIZVAXwmesnj3og | 1805 |
oumi/datasets/vision_language/llava_instruct_mix_vsft.py | sha256=WQu9XeTfr-avyqkopmtBpO8CT7moMFXWBxRTkm-az_w | 6018 |
oumi/datasets/vision_language/mnist_sft.py | sha256=VFE69DkdITqs13letzztdNcqaUiukv33wBSe_ey-w5U | 2648 |
oumi/datasets/vision_language/the_cauldron.py | sha256=VwXXjfSXWKUY8GS-hOC9GTKgkMMhtDG7e-TYzLNPdl4 | 5372 |
oumi/datasets/vision_language/vision_jsonlines.py | sha256=PGNbZGG47Y0IvkwNseiihSs79AOPUaWoZbhdAI4igCY | 4662 |
oumi/datasets/vision_language/vqav2_small.py | sha256=5hHZ516F5CRF6uXJGuyVIg0H5-tjVl4j3ahLU-oTTv8 | 2027 |
oumi/inference/__init__.py | sha256=fz2XED1zc9IpARpAgxNuhIm-tZwqVcS7XuLBJueeEaA | 2290 |
oumi/inference/anthropic_inference_engine.py | sha256=Wnfu89qDxUEiV6aLPvXhRoOALE-wFwtH5-THaZotyOI | 5468 |
oumi/inference/deepseek_inference_engine.py | sha256=Jni2o2W3hheBxfan6p5Lnq1YSNI8mcTfjikJ9O062Ug | 1040 |
oumi/inference/gcp_inference_engine.py | sha256=TQyJA4KCYxTY7WdhPHHqrWQgUzTwiPAUIO1ouk8qBYk | 7570 |
oumi/inference/gemini_inference_engine.py | sha256=YOeLClngotF0TLihG1Z8VwxUa3C4fj_sKYKltnN2NAw | 3388 |
oumi/inference/llama_cpp_inference_engine.py | sha256=sSZu-sxkfYHDnPo1cA9UOInAEP6v5BVC3wxYtGceWv0 | 10565 |
oumi/inference/native_text_inference_engine.py | sha256=Q7nf_59FvIjq6imrciwbIYw7gQVTWm8o58NOvepcwCM | 17081 |
oumi/inference/openai_inference_engine.py | sha256=x3TnYUBbNbc5YGiK6jtZdwE8W7vzAHGRbqzR7pL5Ny0 | 1218 |
oumi/inference/parasail_inference_engine.py | sha256=79TNsZNM1PLiz1xe5OXoUkCY5wRnmVJ1W8Es9LrQtmc | 1229 |
oumi/inference/remote_inference_engine.py | sha256=8YkIrlCLgBnCORr6Uvbn2xiYOS76w2aPY4wd1qhgbz0 | 36106 |
oumi/inference/remote_vllm_inference_engine.py | sha256=q5jIyS9OYYcnwu0kdMEV0MYbR0lPaBf_7Zbtk1A92SI | 4010 |
oumi/inference/sambanova_inference_engine.py | sha256=M0Yy3d5Sx5LxGOuP8DAAaCk1gzMh9wtKbXT2i4AzXXU | 5185 |
oumi/inference/sglang_inference_engine.py | sha256=5fCRwR4sVK_B8i1QjoqD_ENtdq9NGm9QO2WU6iL2rwo | 11697 |
oumi/inference/together_inference_engine.py | sha256=25LduaXgXUJ1edvm451gvJ5dJMkmWVICDB5qRDPAc4g | 1233 |
oumi/inference/vllm_inference_engine.py | sha256=kWJ2qLpL-UcTWo7WHzL1RYbcXCX67P8l_qCBhMXTBZw | 13927 |
oumi/judges/__init__.py | sha256=0R5ptl6dgKeJc7r6DkWXQD5YfHxKdd-79l72uK-3vek | 1663 |
oumi/judges/base_judge.py | sha256=d2ODqFpJS0gOvPZmaEoOeOhfbSnVTGtHb_8dHI53zGc | 9150 |
oumi/judges/judge_court.py | sha256=2Se3U0Q-Sy0P8JRE15rItlz9OuMRpYUGT9iX75KkWLo | 10092 |
oumi/judges/oumi_judge.py | sha256=r7U0BEvEsvYueg8dcnzdX6iblPLV9vpjbVCyQq-3-qg | 3151 |
oumi/judges/oumi_v1/helpful.json | sha256=EFEDDPjdM1LJf74zPmMqiMyjIAnmFRry3J-iG7--5A4 | 3136 |
oumi/judges/oumi_v1/honest.json | sha256=FKrJzgAqOzRCgsV1RMk3Y1LG7FF18UH9Yg7yb-3NahI | 2223 |
oumi/judges/oumi_v1/safe.json | sha256=NYTCwVv-atO5N6CORB92VSRd9VrpDzx9Z9YCJ1gATws | 2736 |
oumi/judges/oumi_v1/valid.json | sha256=Jrf43OtnbOWGPpIdfOeB6uJ4XwdwkXeczGum_lWkZbg | 2195 |
oumi/judges/test_judge/helpful.json | sha256=2t3RVapLJt-6W2FzHDxViJSB32uZLyUaPZjZeuOL3Hs | 994 |
oumi/launcher/__init__.py | sha256=7Dezr6E1QR5602aM3Y2lKgDcA-rUCLcD9i773iyW2SE | 1860 |
oumi/launcher/launcher.py | sha256=W8D0n_9v4gB9_OMH-lpfx0WDTUeLOnfJHl_Uv3efh94 | 6477 |
oumi/launcher/clients/local_client.py | sha256=_sdEh3MURtKkbVIOmw-YaIYnvH96TyF7n6GtwQ3f8D0 | 8832 |
oumi/launcher/clients/polaris_client.py | sha256=O6CZd1cHXoXC2yMwmJob98c-Cc_OoKPUbz0TnLxbCI4 | 14269 |
oumi/launcher/clients/sky_client.py | sha256=G_X75FpbNBBiVBm_LrQtJzFL4OclOcUncysLlHu6vgI | 8378 |
oumi/launcher/clients/slurm_client.py | sha256=1bduZHlaZKlorouCD4onnbUrpK1JVBukjPqzvzOC-sw | 15534 |
oumi/launcher/clouds/__init__.py | sha256=JBOEFaGYz-V_qmt98neqnPLXxaRAck2iOg5Wj-oKG4I | 1702 |
oumi/launcher/clouds/local_cloud.py | sha256=QnqeHkiSWjzpBqZ4HtmClLffAm2uCxRzIVLOlxeCptw | 2906 |
oumi/launcher/clouds/polaris_cloud.py | sha256=QvHFSzU0HxzXtkIMRAg__88WQJNIofc_vTo5HJkS5xc | 5459 |
oumi/launcher/clouds/sky_cloud.py | sha256=f7cAAqEWPnQYlpnoRAr5oGiCpMOK3lN0WfZPTOq_GMw | 4191 |
oumi/launcher/clouds/slurm_cloud.py | sha256=E6Qy1PBgNBjpf0RomgPnwZ9j9iyALma1mIYAish1CMg | 3961 |
oumi/launcher/clusters/local_cluster.py | sha256=vTA4OvVWcYmkwpNVh5bS0iqqPz7PInrAsPN9hp3VYgY | 4601 |
oumi/launcher/clusters/polaris_cluster.py | sha256=6GvWyO3tdvOpOuKmHwKUGVDDIa6E_HfNl5jt_LqTmb4 | 11102 |
oumi/launcher/clusters/sky_cluster.py | sha256=Yk-iGsEEXlJXOhoHwrAKYJiseZ3LRiojkywiL_lLDuo | 3640 |
oumi/launcher/clusters/slurm_cluster.py | sha256=XO5y3F2zM-YDrJCuR7x-hirUY5I0qQa4kiOrHVEf93A | 10310 |
oumi/models/__init__.py | sha256=vD9kwj6NlSuvKwR3-5hIwbVWplLelcJG1MX_ZSv1d1s | 1694 |
oumi/models/cnn_classifier.py | sha256=SduubAEthYmpPASy-KHwQEd7sH_mTVGf_J--kIpCQsM | 3988 |
oumi/models/mlp.py | sha256=9I5vEZoUE7_jweRVPz6lXOCR3gchjwsuiXddioBQsjY | 3148 |
oumi/models/experimental/cambrian/__init__.py | sha256=QJW6BGfD_R3Hk92fNAqJLISj3QlRcWQOhIdegQ6k_XI | 103 |
oumi/models/experimental/cambrian/constants.py | sha256=BcqlMgYPYbMMraW6I9oPS_ZAyJgpiRP3aGU1g0guWQE | 335 |
oumi/models/experimental/cambrian/mm_utils.py | sha256=4lcqE-aYiV9qfSCCk3XAYrRFICvLXBc1bo4tisEpuC8 | 11604 |
oumi/models/experimental/cambrian/utils.py | sha256=W3heWR6PwYX2dBMrKsAx_8AI1X4Z4n4-mnSoezq0QRk | 446 |
oumi/models/experimental/cambrian/model/__init__.py | sha256=v9o8WP89t-ND8IiGfbt-svKZdy5yYbjGFqECbu6jvu8 | 84 |
oumi/models/experimental/cambrian/model/builder.py | sha256=Kj8CWkmQqQKE-VsZmk9tkHEd5j3HZX8BxsyblaZg3Oo | 9375 |
oumi/models/experimental/cambrian/model/cambrian_arch.py | sha256=syz83k8mJm72h9tGmLB49OmJoshc5sv2NLkPrhpT_4s | 46711 |
oumi/models/experimental/cambrian/model/vision_sampler.py | sha256=RzsFydR7j587IIzb0_8HtTqhqgAM4QIOXiQtZpWd4iw | 19407 |
oumi/models/experimental/cambrian/model/language_model/cambrian_llama.py | sha256=WVmoDR8dX2vzJUIsMPO6wGJdof8XRy1k04oTyrUpk7A | 26077 |
oumi/models/experimental/cambrian/model/language_model/cambrian_phi3.py | sha256=xZj3mNp6e-Aqu0qrkM8gUFxW_7Rv8MrK2isVdpXt5so | 8661 |
oumi/models/experimental/cambrian/model/language_model/phi3/__init__.py | sha256=kXlEMVOo6S07hVtkv9w-vRRPs5jC1cqkeTFgMH1ytvc | 237 |
oumi/models/experimental/cambrian/model/multimodal_encoder/__init__.py | sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU | 0 |
oumi/models/experimental/cambrian/model/multimodal_encoder/base_encoder.py | sha256=Bo1QKqW1zI4H-0tZG7AWzp4sY95CpfV951odbTyuQyc | 4137 |
oumi/models/experimental/cambrian/model/multimodal_encoder/builder.py | sha256=MoaESF_DidLqalehDFp-kMh5PuFlbi75r25TVWdLARU | 3804 |
oumi/models/experimental/cambrian/model/multimodal_encoder/clip_convnext_encoder.py | sha256=RY9rIA4RbWCXYYwH1y4rxqSJDkGYwt1Xms2ydBcVccY | 6722 |
oumi/models/experimental/cambrian/model/multimodal_encoder/clip_encoder.py | sha256=K28EBw0l2wBH8wdFr042X7BBMxw4zp1eL_dcO6I90WQ | 4397 |
oumi/models/experimental/cambrian/model/multimodal_encoder/dino_encoder.py | sha256=i6df9sAJ8cro9Bk2jzwohBXfiXmHEWawyLs4A4nlZ5A | 7212 |
oumi/models/experimental/cambrian/model/multimodal_encoder/load.py | sha256=HAlXB9bZ8tZCNqhg_BuBfz1xpvB_NBWXVyIWd7O4Sd8 | 1810 |
oumi/models/experimental/cambrian/model/multimodal_encoder/siglip_encoder.py | sha256=ORVG3TAGN0X7ACfRozYHYFGjOIjoo3kROMQ2LStarBk | 3953 |
oumi/models/experimental/cambrian/model/multimodal_projector/builder.py | sha256=4Vm27QQz9qdaykL5dR0HL93WUJoPOmzuwFGMm-ApiJQ | 2318 |
oumi/models/experimental/cambrian/model/multimodal_projector/projectors.py | sha256=X3RzNKNMYnq95gqKriuJdX21UIAV1GYLGO3bHxxWXlE | 4457 |
oumi/models/layers/ring_attention.py | sha256=Ra4TF3OzFMyaFAiCX8_nQwWMsaW4AjsYcxfZPdBCgwg | 5812 |
oumi/models/layers/zigzag.py | sha256=iCELXA_mqO3J5N1rLWrUsZzMRbwmVx33klifqxoLLD0 | 12571 |
oumi/models/layers/zigzag_utils.py | sha256=e43rbN6yndazt6_8C3eW8OimqA8G2w-AbWZSOk9Se-k | 5556 |
oumi/performance/mfu.py | sha256=yE-wAr62lSMyYl8i0CoxXNptW7MX_4-AoNTvlSFXg3E | 6257 |
oumi/performance/telemetry.py | sha256=wikxLN9GEbQIkssaCS3RGskXLPqnX7z7geo42t0Yne8 | 20088 |
oumi/performance/torch_profiler_utils.py | sha256=KzxRzWXzYgc2Ru13TOxyjCtUGnfbLu9fUmz2-vh0y2M | 7957 |
oumi/utils/batching.py | sha256=vJZ6F2ZSgunYCXBORIfDz3eBmLeNfZumfCuucz6gwyc | 1440 |
oumi/utils/conversation_utils.py | sha256=OWYpL83NtwY--TNKyT2d3zHq_Hvtd6bEj0Pe8bvT5bo | 8951 |
oumi/utils/device_utils.py | sha256=Oi9VZmSI7GsHZMzkXu5-if-8VXhTUbjHvtm8IngIYGA | 12806 |
oumi/utils/distributed_utils.py | sha256=NoTB0IR2NVfnPFIaw37PnjhAX1tsP9cq88R0CA2U5gY | 1425 |
oumi/utils/git_utils.py | sha256=tdkkTGJzQtV5fgFjaI-5nSHQIx_Ovm0GHitSKIMQ4xY | 2180 |
oumi/utils/hf_utils.py | sha256=_OtWhOgqqaLWWk-JDVYFqohnc7dui7T4LwlWmZbkunQ | 3441 |
oumi/utils/image_utils.py | sha256=ZJ799RNXF_Iuafik3Uz6Covik386pj3lEBANIg0u0ks | 12559 |
oumi/utils/io_utils.py | sha256=a9FNnDRWhNKToTFmxpyGBJ-Mtr8Im4zEm74aKQc0GI0 | 3959 |
oumi/utils/logging.py | sha256=7ZnICbobn9IFCafkxZjT7jN-QepN0h84d-zgxHXyfNE | 4649 |
oumi/utils/model_caching.py | sha256=lOh_C6gY4HPunfge3kbRdB1Avh2jIU-bxWE9jBFKS54 | 2419 |
oumi/utils/packaging.py | sha256=KCa4t-gdlq4gHRPCcScyFIOE-r1DbzahKh-pCpt3L5A | 6243 |
oumi/utils/peft_utils.py | sha256=hT7ufLteAHLME0SRLfQtYORJYup4KsNxagO3RaEI2Ew | 1721 |
oumi/utils/saver.py | sha256=Ai4-VhEr8ViaKI4tedT4cpkEfXLTe6lu_JHUnqQmgwA | 6184 |
oumi/utils/serialization_utils.py | sha256=e8yON_kIcnObH5aQ9Vqfzz19leu9WXvI-lkyoYtZxL0 | 3192 |
oumi/utils/str_utils.py | sha256=U7ER0WiHHoGjoIDsFxFIMZK52JmfBTI_JR6YNjEERJ4 | 3851 |
oumi/utils/torch_naming_heuristics.py | sha256=sxM3MITukc_85p_gcEM7rBl_wTlg-pEqwOSuA74hzyI | 5203 |
oumi/utils/torch_utils.py | sha256=KSjO3I5pG1zq1FJJ7-tD-cRgxUkEUfNOfefG0C81BlQ | 25752 |
oumi/utils/version_utils.py | sha256=3DRtDvMcKn83X8mbgPySgKxr3bxqelhtO880Te7wQoE | 1154 |
oumi-0.1.8.dist-info/LICENSE | sha256=HKM7ZIBgTv6ilauIaySY-7L8KpakTcj41IWMTDZYaio | 11336 |
oumi-0.1.8.dist-info/METADATA | sha256=tuhUzMDIS4sENnJHgBQZIFy3oMNkOQu9OcUq1jlhE_o | 42973 |
oumi-0.1.8.dist-info/WHEEL | sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8 | 91 |
oumi-0.1.8.dist-info/entry_points.txt | sha256=uwxG13ONpj19TkOJjLdeGhaTvmXymhYj-aMtTiDCh-A | 43 |
oumi-0.1.8.dist-info/top_level.txt | sha256=0WhALTeKKEPgXWlKlYatvKtTG47SAatuqX-yAcNyTCk | 18 |
oumi-0.1.8.dist-info/RECORD | — | — |
top_level.txt
experimental
oumi
entry_points.txt
oumi = oumi.cli.main:run