build: Moved ibm-fms to optional dependecy aiu

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

README.md

@@ -104,6 +104,7 @@ The following optional dependencies are available:
 - `gptq`: `GPTQModel` package for W4A16 quantization
 - `mx`: `microxcaling` package for MX quantization
 - `opt`: Shortcut for `fp8`, `gptq`, and `mx` installs
+- `aiu`: `ibm-fms` package for AIU model deployment
 - `torchvision`: `torch` package for image recognition training and inference
 - `triton`: `triton` package for matrix multiplication kernels
 - `examples`: Dependencies needed for examples

pyproject.toml

@@ -30,7 +30,7 @@ dependencies = [
 "datasets>=3.0.0,<4.0",
 "pandas",
 "safetensors",
-"ibm-fms>=0.0.8",
+
 "pkginfo>1.10",
 ]
 
@@ -40,6 +40,7 @@ fp8 = ["llmcompressor"]
 gptq = ["Cython", "gptqmodel>=1.7.3"]
 mx = ["microxcaling>=1.1"]
 opt = ["fms-model-optimizer[fp8, gptq, mx]"]
+aiu = ["ibm-fms>=0.0.8"]
 torchvision = ["torchvision>=0.17"]
 flash-attn = ["flash-attn>=2.5.3,<3.0"]
 triton = ["triton>=3.0,<3.4"]