callmeclover commited on
Commit
f8674c2
1 Parent(s): d3137af

Update setup.py

Browse files
Files changed (1) hide show
  1. setup.py +3 -59
setup.py CHANGED
@@ -1,61 +1,5 @@
1
- import pathlib
2
- import setuptools
3
- import pkg_resources
4
 
 
5
 
6
- with open("README.md", "r") as fh:
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")