From 937d3a3ae45c10f7c5d5d1e5f07f03c736e87922 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 24 Jan 2024 15:40:14 -0500 Subject: [PATCH] [README] --- README.md | 2 +- pyproject.toml | 16 ++++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index 72bb8d9..d1b8b57 100644 --- a/README.md +++ b/README.md @@ -9,7 +9,7 @@ Implementation of Switch Transformers from the paper: "Switch Transformers: Scal ## Installation ```bash -pip install -e . +pip install switch-transformers ``` # Usage diff --git a/pyproject.toml b/pyproject.toml index 5d4ac8e..8c2018e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -3,15 +3,15 @@ requires = ["poetry-core>=1.0.0"] build-backend = "poetry.core.masonry.api" [tool.poetry] -name = "paper" +name = "switch-transformers" version = "0.0.1" -description = "Paper - Pytorch" +description = "SwitchTransformers - Pytorch" license = "MIT" authors = ["Kye Gomez "] -homepage = "https://github.com/kyegomez/paper" -documentation = "https://github.com/kyegomez/paper" # Add this if you have documentation. +homepage = "https://github.com/kyegomez/SwitchTransformers" +documentation = "https://github.com/kyegomez/SwitchTransformers" # Add this if you have documentation. readme = "README.md" # Assuming you have a README.md -repository = "https://github.com/kyegomez/paper" +repository = "https://github.com/kyegomez/SwitchTransformers" keywords = ["artificial intelligence", "deep learning", "optimizers", "Prompt Engineering"] classifiers = [ "Development Status :: 4 - Beta", @@ -25,9 +25,9 @@ classifiers = [ python = "^3.6" swarms = "*" zetascale = "*" - -[tool.poetry.dev-dependencies] -# Add development dependencies here +einops = "*" +torch = "*" +torchvision = "*" [tool.poetry.group.lint.dependencies]