aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAnthony Wang2022-02-26 12:06:21 -0600
committerAnthony Wang2022-02-26 12:06:21 -0600
commit1a13ba99092e55d5e236be46a7bf7f20ea57f599 (patch)
tree88aca36d9a8a237904137f34de7107f513b78f0e
parent123846ed7dd429387e069c186bb4d47c24eb8124 (diff)
Revert "Fix pip command to install pytorch instead of old torch package"
This reverts commit 123846ed7dd429387e069c186bb4d47c24eb8124.
-rw-r--r--README.md2
1 files changed, 1 insertions, 1 deletions
diff --git a/README.md b/README.md
index 38b7218..f10a28f 100644
--- a/README.md
+++ b/README.md
@@ -4,7 +4,7 @@ Fediverse ebooks bot using neural networks
## Usage
-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.
+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.
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.