a/ebooks
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix pip command to install pytorch instead of old torch package

Browse source
This commit is contained in:
Anthony Wang 2022-02-26 12:02:44 -06:00
parent 4a9671c958
commit 123846ed7d
Signed by: a
GPG key ID: BC96B00AEC5F2D76
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -4,7 +4,7 @@ Fediverse ebooks bot using neural networks
## Usage
First, install Python dependencies with `pip install psycopg2 torch transformers datasets Mastodon.py`. Currently only Mastodon with PostgreSQL is supported, but it should be easy to add support for other platforms and databases.
First, install Python dependencies with `pip install psycopg2 pytorch transformers datasets Mastodon.py`. Currently only Mastodon with PostgreSQL is supported, but it should be easy to add support for other platforms and databases.
Generate the training data from the Mastodon database using `python data.py -d 'dbname=test user=postgres password=secret'`. You can skip this step if you have training data from another source.