Wheelodex » Projects » oat-llm

oat-llm

View on PyPI — Reverse Dependencies (0)

0.0.4 oat_llm-0.0.4-py3-none-any.whl

Wheel Details

Project: oat-llm
Version: 0.0.4
Filename: oat_llm-0.0.4-py3-none-any.whl
Download: [link]
Size: 83708
MD5: 21a8861e8e6d7369bdade59525bd31df
SHA256: 14fe63e9b2defbcff4bd38e3078350fe0523fa2704ed41f4cd4304520bc3dc99
Uploaded: 2024-11-11 09:26:07 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: oat-llm
Version: 0.0.4
Summary: Online AlignmenT (OAT) for LLMs.
Author: Zichen Liu
Author-Email: Zichen Liu <liuzc[at]sea.com>, Changyu Chen <chency[at]sea.com>
Home-Page: https://github.com/sail-sg/oat
Project-Url: Documentation, https://github.com/sail-sg/oat#readme
Project-Url: Issues, https://github.com/sail-sg/oat/issues
Project-Url: Source, https://github.com/sail-sg/oat
License: Apache-2.0
Keywords: rlhf,llm,ai-alignment,rl,bandit,ai,sample-efficiency
Classifier: Development Status :: 4 - Beta
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: Implementation :: CPython
Classifier: Programming Language :: Python :: Implementation :: PyPy
Requires-Python: >=3.8, <3.11
Requires-Dist: accelerate
Requires-Dist: bitsandbytes
Requires-Dist: datasets
Requires-Dist: deepspeed (==0.15.0)
Requires-Dist: dm-launchpad[tensorflow]
Requires-Dist: einops
Requires-Dist: flash-attn (==2.6.3)
Requires-Dist: fire
Requires-Dist: httpx
Requires-Dist: jsonlines
Requires-Dist: llm-blender (==0.0.2)
Requires-Dist: loralib
Requires-Dist: ml-collections
Requires-Dist: mosec
Requires-Dist: msgspec
Requires-Dist: optimum
Requires-Dist: packaging
Requires-Dist: peft
Requires-Dist: protobuf (==3.20.0)
Requires-Dist: pyarrow (<12,>=0.6.1)
Requires-Dist: python-Levenshtein (==0.25.1)
Requires-Dist: scipy
Requires-Dist: torchmetrics
Requires-Dist: tqdm
Requires-Dist: tyro
Requires-Dist: transformers (==4.45.2)
Requires-Dist: transformers-stream-generator
Requires-Dist: vllm (==0.6.2)
Requires-Dist: wandb
Requires-Dist: wheel
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 9854 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.43.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
oat/__about__.py sha256=lsz5Tv78yL8QfG1xLSvA_VgO1qtQjpdrANCrcnZ9pK0 631
oat/__init__.py sha256=1qFdHw5d-gq-rUaVGMqwzZxyExAbrz4XJ6jQNrqu450 594
oat/actor.py sha256=LA9siXZN-DXqfGXMKuyRU9ml5O0aafUPBeS6PJPbLBE 14259
oat/args.py sha256=7QHPjVJaCcVHGpQIVcugEXBJj5jxU78QgCTaPyt3rYM 7001
oat/exploration.py sha256=gl1vnvnBMeREy_jQ1UDl_LPka-XOlo4MgGDtD6FVWwc 14776
oat/interface.py sha256=19-ci_BxTGNkG5ssCAkiUWyyxx9Ff8Q7P6viecVDrF4 3779
oat/model.py sha256=XkBu_msRRalzimEnwx-aBvuHj8WTIWl1XdUOuUdL1qM 5548
oat/types.py sha256=zUuFDGTsXCNGMcgO5iv2cLuWb44FOjAVjKbPcM1rA64 1235
oat/baselines/__init__.py sha256=1qFdHw5d-gq-rUaVGMqwzZxyExAbrz4XJ6jQNrqu450 594
oat/baselines/apl.py sha256=USXtiuJHdMnuLN-tsmyK2rBIufhbmqUHaMNXtB1orWw 15822
oat/baselines/xpo.py sha256=KFiymBnp_yv-0GODkL0yfySjjmS8Rj5woW4Tn-k28Xk 4803
oat/experiment/__init__.py sha256=1qFdHw5d-gq-rUaVGMqwzZxyExAbrz4XJ6jQNrqu450 594
oat/experiment/main.py sha256=IfcwsySyZ6Jp9P8pRdXAhvsLL5h5n5a1F-MU7i0ypmI 1286
oat/experiment/offline_lp.py sha256=tb7jtmdlFzflX4R3iT6--ZXlTFBNgLnVZeFYKJArTJw 1500
oat/experiment/run_apl.py sha256=b_5Yu713QgdSHZWQfjrkWli-59l_FmM-f1SnqxYQCjM 1509
oat/experiment/run_xpo.py sha256=1vko4lQbUgcfmf-01ki3r5EnMODxZhqKSuIq4Kw6eZ4 1343
oat/learners/__init__.py sha256=9QC1-F8__39vXwxu9lIvpr3jbf1nWj7Hkj9Wts9USp8 805
oat/learners/base.py sha256=97BzUInenxmAmTl9W8jBpndkMGSpRp9z0WtGnLifBO0 24791
oat/learners/dap.py sha256=DpftQ2AmrT_pqmhg91IzpWoj_7n4efsB97k130BZLaI 7050
oat/learners/dap_with_rm.py sha256=0vns4KDPMxVbnQy1GVlcuRDgSpRonIr9DwryUIFm_a4 4808
oat/learners/loss.py sha256=7_7mn7-CQrCNyb-mKMfrSJkqYEuTxFJlCPgSZ0MOJqk 3995
oat/learners/offline_dap.py sha256=xnetkalLRlHA_ekX4wRFQoelx0cub53v3HfrVDdgUAg 4756
oat/oracles/__init__.py sha256=fTtbSYABQqFHffahwGkfxjoObZtxDJefFNEm1rqL7dg 997
oat/oracles/base.py sha256=TQmuhAMUVoa28E1Ibk6glnT74sBSMyhQWl3atujZmx4 1599
oat/oracles/gpt.py sha256=9wZFN23gW8s_obWTIxBUI3esXMCa7Txf0hISvw5bh-8 6195
oat/oracles/pair.py sha256=XEMaODKR6AU8sL1rTfKWUVFXOgkzYqIskM8t3EvisA0 1522
oat/oracles/remote/__init__.py sha256=1qFdHw5d-gq-rUaVGMqwzZxyExAbrz4XJ6jQNrqu450 594
oat/oracles/remote/client.py sha256=BHHMR31wqbEwzAZye35MJbMBxXVkLRR1OAF4OYJQEKU 3661
oat/oracles/remote/server.py sha256=_PdPEgW-qrn2Lgw3PrwvQT0EcSanZCM6t71egIEIFvE 4412
oat/rm/__init__.py sha256=1qFdHw5d-gq-rUaVGMqwzZxyExAbrz4XJ6jQNrqu450 594
oat/rm/backbone.py sha256=kdD3fRmpZL9RTuMPGNRAT0LygCsv3S9g2ZRZannOc5w 8514
oat/rm/model.py sha256=UNEsfLruQQzpAT3d_BDknEEJu67knelN1fq9kIHEgPQ 10253
oat/rm/networks.py sha256=t_hEFCIvWOb2blZnqGhWDUtc7szu_nDEs99hINnmj3k 4628
oat/rm/uncertainty.py sha256=awxl0B2_oSOdK-6L1OICIvztW-Ajvel33PM9Xrg2II0 2802
oat/utils/__init__.py sha256=1qFdHw5d-gq-rUaVGMqwzZxyExAbrz4XJ6jQNrqu450 594
oat/utils/buffer.py sha256=UzLam3S9fR9VJdKyJ5eS1KqL0t9A_qmcH21QzGubRZw 2195
oat/utils/data.py sha256=HHxGEdyS3pPYeq3ea0CPFzYI7LbNHjqUHogyxNg0N1A 13986
oat/utils/deepspeed.py sha256=iGay-LzOrKIAbB-FbQk6ij8-3rCYVW4mRcb-EiqIvZc 22800
oat/utils/distributed.py sha256=IpQNOGNqf3QMFr0J8RYCyLMXXu1kwZ8kymJEwFTI-Xg 6367
oat/utils/ipc.py sha256=1t-f6TxsXVdHlN_rtF9IUojGyXZ7bSPr9DuKuUS2r54 2969
oat/utils/launcher.py sha256=_0e164FG8rSFUsyPeksLOQSFiRgcQeR5w6YJCcojJ2g 1574
oat_llm-0.0.4.dist-info/LICENSE sha256=ITvk9LYfj913BttV6IpAgfbDx-tp10FTgQOtZMXDH5k 11358
oat_llm-0.0.4.dist-info/METADATA sha256=VOu6pNGYrPJXX0lYW1XddcQreYlItHumGaVybCJ0xrc 11651
oat_llm-0.0.4.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
oat_llm-0.0.4.dist-info/top_level.txt sha256=DDLx4L-7a-XNurRXMHbz5QX0avtmuu3x7jR0E4wTCmk 4
oat_llm-0.0.4.dist-info/RECORD — —

top_level.txt

oat