Merge pull request #5 from asappresearch/remove_bits_and_bytes

Remove bits and bytes

Author: blattimer-asapp

README.md

@@ -11,11 +11,15 @@ This repo runs JOSH, the ToolWOZ, and τ-bench dataset. This repo also contains
 ## Setup
 1. Run the following in a new env
 ```
+pip install josh-train
+```
+or
+```
 pip install -e .
 ```
-2. Unzip the ```dataset.zip``` file in the ```data``` folder
+1. Unzip the ```dataset.zip``` file in the ```data``` folder
 
-3. Set up your openai credentials
+2. Set up your openai credentials
 ```
 export OPENAI_API_KEY= # api_key
 export OPENAI_ORGANIZATION= # api_org

requirements.txt

@@ -6,7 +6,6 @@ pyyaml
 datasets
 accelerate
 transformers
-bitsandbytes
 numpy
 pandas
 trl

setup.py

@@ -25,7 +25,7 @@ def get_requires(requires_filename: str) -> List[str]:
     install_requires=get_requires("requirements.txt"),
     include_package_data=True,
     setup_requires=["setuptools_scm"],
-    version="0.1.0",
+    version="0.1.1",
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",