|
1 | | -from setuptools import setup, find_packages |
| 1 | +from setuptools import setup |
2 | 2 |
|
3 | | -long_description = ''' |
| 3 | +long_description = """ |
4 | 4 | A simple Python package that wraps existing model fine-tuning and generation scripts for OpenAI GPT-2 text generation model (specifically the "small", 124M hyperparameter version). Additionally, this package allows easier generation of text, generating to a file for easy curation, allowing for prefixes to force the text to start with a given phrase. |
5 | 5 |
|
6 | 6 | ## Usage |
|
41 | 41 | You can pass a `run_name` parameter to `finetune` and `load_gpt2` if you want to store/load multiple models in a `checkpoint` folder. |
42 | 42 |
|
43 | 43 | NB: *Restart the Python session first* if you want to finetune on another dataset or load another model. |
44 | | -''' |
| 44 | +""" |
45 | 45 |
|
46 | 46 |
|
47 | 47 | setup( |
48 | | - name='gpt_2_simple', |
49 | | - packages=['gpt_2_simple'], # this must be the same as the name above |
50 | | - version='0.7.1', |
51 | | - description="Python package to easily retrain OpenAI's GPT-2 " \ |
| 48 | + name="gpt_2_simple", |
| 49 | + packages=["gpt_2_simple"], # this must be the same as the name above |
| 50 | + version="0.7.2", |
| 51 | + description="Python package to easily retrain OpenAI's GPT-2 " |
52 | 52 | "text-generating model on new texts.", |
53 | 53 | long_description=long_description, |
54 | | - long_description_content_type='text/markdown', |
55 | | - author='Max Woolf', |
56 | | - |
57 | | - url='https://github.com/minimaxir/gpt-2-simple', |
58 | | - keywords=['deep learning', 'tensorflow', 'text generation'], |
| 54 | + long_description_content_type="text/markdown", |
| 55 | + author="Max Woolf", |
| 56 | + |
| 57 | + url="https://github.com/minimaxir/gpt-2-simple", |
| 58 | + keywords=["deep learning", "tensorflow", "text generation"], |
59 | 59 | classifiers=[], |
60 | | - license='MIT', |
| 60 | + license="MIT", |
61 | 61 | entry_points={ |
62 | | - 'console_scripts': ['gpt_2_simple=gpt_2_simple.gpt_2:cmd'], |
| 62 | + "console_scripts": ["gpt_2_simple=gpt_2_simple.gpt_2:cmd"], |
63 | 63 | }, |
64 | | - python_requires='>=3.5', |
| 64 | + python_requires=">=3.6", |
65 | 65 | include_package_data=True, |
66 | | - install_requires=['regex', 'requests', 'tqdm', 'numpy', 'toposort'] |
| 66 | + install_requires=["regex", "requests", "tqdm", "numpy", "toposort"], |
67 | 67 | ) |
0 commit comments