Merge pull request #84 from ericjang/master

Add setup.py to allow mingpt to be used as a third-party library

README.md

@@ -12,6 +12,16 @@ The minGPT library is three files: [mingpt/model.py](mingpt/model.py) contains t
 - `demo.ipynb` shows a minimal usage of the `GPT` and `Trainer` in a notebook format on a simple sorting example
 - `generate.ipynb` shows how one can load a pretrained GPT2 and generate text given some prompt
 
+### Library Installation
+
+If you want to `import mingpt` into your project:
+
+```
+git clone https://github.com/karpathy/minGPT.git
+cd minGPT
+pip install -e .
+```
+
 ### Usage
 
 Here's how you'd instantiate a GPT-2 (124M param version):

setup.py

@@ -0,0 +1,12 @@
+from setuptools import setup
+
+setup(name='minGPT',
+      version='0.0.1',
+      author='Andrej Karpathy',
+      packages=['mingpt'],
+      description='A PyTorch re-implementation of GPT',
+      license='MIT',
+      install_requires=[
+            'torch',
+      ],
+)