Skip to content

Commit f10b56a

Browse files
committed
build: Moved ninja, evaluate, and huggingface_hub to examples optional dep
Signed-off-by: Brandon Groth <[email protected]>
1 parent a17b499 commit f10b56a

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

fms_mo/utils/import_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
"fms",
3333
"triton",
3434
"torchvision",
35+
"huggingface_hub",
3536
]
3637

3738
available_packages = {}

pyproject.toml

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,21 +25,17 @@ dependencies = [
2525
"numpy>=1.26.4,<2.3.0",
2626
"accelerate>=0.20.3,!=0.34,<1.7",
2727
"transformers>=4.45,<4.53",
28-
"torch>=2.2.0,<2.6",
28+
"torch>=2.2.0,<2.6",
2929
"tqdm>=4.66.2,<5.0",
3030
"datasets>=3.0.0,<4.0",
31-
"ninja>=1.11.1.1,<2.0",
32-
"tensorboard",
33-
"notebook",
34-
"evaluate",
35-
"huggingface_hub",
3631
"pandas",
3732
"safetensors",
3833
"ibm-fms>=0.0.8",
3934
"pkginfo>1.10",
4035
]
4136

4237
[project.optional-dependencies]
38+
examples = ["ninja>=1.11.1.1,<2.0", "evaluate", "huggingface_hub"]
4339
fp8 = ["llmcompressor"]
4440
gptq = ["Cython", "gptqmodel>=1.7.3"]
4541
mx = ["microxcaling>=1.1"]

0 commit comments

Comments
 (0)