Skip to content

Commit

Permalink
README.md updates and version bump
Browse files Browse the repository at this point in the history
  • Loading branch information
rajcscw committed Nov 23, 2022
1 parent 7ea9c3e commit 6ee301c
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 8 deletions.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,12 @@ Thoroughly **tested** and **benchmarked** with over **2000 experiments** :fire:

All of these building blocks can be customizable allowing users to train transformer-based LMs to optimize any arbitrary reward function on any dataset of their choice.

## Recent updates (v0.2.0) on 23-Nov-22
- Added daily dialog task
- Fixed compatibility issues with some Seq2seq models such as BART, blendorbot etc
- Implemented data parallel support
- Refactored policy classes

---
# Install

Expand Down
18 changes: 10 additions & 8 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,13 @@
with open("requirements.txt") as fp:
requirements = fp.read().splitlines()

setup(name="rl4lms",
version="0.1.0",
description="A library for training language models (LM) using RL",
author="Rajkumar Ramamurthy, Prithviraj Ammanabrolu",
packages=find_packages(),
python_requires='>=3.7',
install_requires=requirements,
url="https://github.com/allenai/RL4LMs")
setup(
name="rl4lms",
version="0.2.0",
description="A library for training language models (LM) using RL",
author="Rajkumar Ramamurthy, Prithviraj Ammanabrolu",
packages=find_packages(),
python_requires=">=3.7",
install_requires=requirements,
url="https://github.com/allenai/RL4LMs",
)

0 comments on commit 6ee301c

Please sign in to comment.