/gym-LLM

OpenAI Gym for training LLMs (large language models)

Primary LanguagePython

Gym LLM


Installation

Install OpenAi Gym

pip install gym

Download and install gym-LLM

git clone https://github.com/Cybernetic1/gym-LLM.git
cd gym-LLM
python setup.py install

Running

Start by importing the package and initializing the environment

import gym
import gym_LLM
env = gym.make('LLM-v1') 

Settings

You can change the rewards by editing the settings.xml placed in your gym-LLM installation folder..

Acknowledgement

Modified from Gym TicTacToe by Clement Romac.