From 6ee301cda0af0533b9714409b2feb116646dfad8 Mon Sep 17 00:00:00 2001 From: Rajkumar Date: Wed, 23 Nov 2022 09:17:59 +0000 Subject: [PATCH] README.md updates and version bump --- README.md | 6 ++++++ setup.py | 18 ++++++++++-------- 2 files changed, 16 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index 050e5aef..24b48778 100644 --- a/README.md +++ b/README.md @@ -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 diff --git a/setup.py b/setup.py index 8a726d64..bb351ce0 100644 --- a/setup.py +++ b/setup.py @@ -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", +)