Add amd extra to setup.py (#1567)

huggingface · Dec 6, 2023 · 71a2a63 · 71a2a63
1 parent c0230e0
commit 71a2a63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -79,6 +79,7 @@
     "neuron": "optimum-neuron[neuron]",
     "neuronx": "optimum-neuron[neuronx]",
     "furiosa": "optimum-furiosa",
+    "amd": "optimum-amd",
     "dev": TESTS_REQUIRE + QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,
     "quality": QUALITY_REQUIRE,