Move tpu extras to the mandatory deps

huggingface · Apr 3, 2024 · b24897b · b24897b
1 parent bcf6080
commit b24897b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,12 +41,11 @@ keywords = [
     "google-cloud"
 ]
 
-dependencies = ["transformers == 4.38.1", ]
+dependencies = ["transformers == 4.38.1", "torch-xla>=2.2.0", "torch>=2.2.0"]
 
 [project.optional-dependencies]
 tests = ["pytest", "safetensors"]
 quality = ["black", "ruff", "isort",]
-tpu = ["wheel", "torch-xla>=2.2.0", "torch>=2.2.0",]
 
 [project.urls]
 Homepage = "https://hf.co/hardware"