Commit
·
f8674c2
1
Parent(s):
d3137af
Update setup.py
Browse files
setup.py
CHANGED
@@ -1,61 +1,5 @@
|
|
1 |
-
import
|
2 |
-
import setuptools
|
3 |
-
import pkg_resources
|
4 |
|
|
|
5 |
|
6 |
-
|
7 |
-
long_description = fh.read()
|
8 |
-
|
9 |
-
with pathlib.Path('requirements.txt').open() as requirements_txt:
|
10 |
-
install_requires = [
|
11 |
-
str(requirement)
|
12 |
-
for requirement
|
13 |
-
in pkg_resources.parse_requirements(requirements_txt)
|
14 |
-
]
|
15 |
-
|
16 |
-
setuptools.setup(
|
17 |
-
name="python-transformer-chatbots-butyr",
|
18 |
-
version="0.1.2",
|
19 |
-
package_dir={"": "src"},
|
20 |
-
packages=setuptools.find_namespace_packages(where="src"),
|
21 |
-
author="Leonid Butyrev",
|
22 |
-
author_email="L.Butyrev@gmx.de",
|
23 |
-
description="A collection of transformer based chatbots.",
|
24 |
-
long_description=long_description,
|
25 |
-
long_description_content_type="text/markdown",
|
26 |
-
url="https://github.com/butyr/python-transformer-chatbot",
|
27 |
-
classifiers=[
|
28 |
-
"Programming Language :: Python :: 3",
|
29 |
-
"License :: OSI Approved :: GPL 2.0 License",
|
30 |
-
"Operating System :: OS Independent",
|
31 |
-
],
|
32 |
-
python_requires='>=3.7',
|
33 |
-
install_requires=[
|
34 |
-
'certifi==2021.5.30',
|
35 |
-
'charset-normalizer==2.0.6',
|
36 |
-
'click==8.0.1',
|
37 |
-
'filelock==3.1.0',
|
38 |
-
'huggingface-hub==0.0.17',
|
39 |
-
'idna==3.2',
|
40 |
-
'importlib-metadata==4.8.1',
|
41 |
-
'joblib==1.0.1',
|
42 |
-
'numpy==1.21.2',
|
43 |
-
'packaging==21.0',
|
44 |
-
'Pillow==9.0.1',
|
45 |
-
'pyparsing==2.4.7',
|
46 |
-
'PyYAML==5.4.1',
|
47 |
-
'regex==2021.9.24',
|
48 |
-
'requests==2.26.0',
|
49 |
-
'sacremoses==0.0.46',
|
50 |
-
'six==1.16.0',
|
51 |
-
'tokenizers==0.10.3',
|
52 |
-
'torch==1.9.1',
|
53 |
-
'torchaudio==0.9.1',
|
54 |
-
'torchvision==0.10.1',
|
55 |
-
'tqdm==4.62.3',
|
56 |
-
'transformers==4.11.0',
|
57 |
-
'typing-extensions==3.10.0.2',
|
58 |
-
'urllib3==1.26.7',
|
59 |
-
'zipp==3.5.0',
|
60 |
-
]
|
61 |
-
)
|
|
|
1 |
+
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
|
|
|
2 |
|
3 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
|
4 |
|
5 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/blenderbot-400M-distill")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|