You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: intel-gaudi-backend-for-tgi.md
+13-8Lines changed: 13 additions & 8 deletions
Original file line number
Diff line number
Diff line change
@@ -11,7 +11,8 @@ We're excited to announce the native integration of Intel Gaudi hardware support
11
11
12
12
## β¨ What's New?
13
13
14
-
We've fully integrated Gaudi support into TGI's main codebase. Previously, we maintained a separate fork for Gaudi devices at [tgi-gaudi](https://github.com/huggingface/tgi-gaudi). This was cumbersome for users and prevented us from supporting the latest TGI features at launch. Now using the new [TGI multi-backend architecture](https://huggingface.co/blog/tgi-multi-backend), we support Gaudi directly on TGI no more finicking on a custom repository π
14
+
We've fully integrated Gaudi support into TGI's main codebase in PR [#3091](https://github.com/huggingface/text-generation-inference/pull/3091). Previously, we maintained a separate fork for Gaudi devices at [tgi-gaudi](https://github.com/huggingface/tgi-gaudi). This was cumbersome for users and prevented us from supporting the latest TGI features at launch. Now using the new [TGI multi-backend architecture](https://huggingface.co/blog/tgi-multi-backend), we support Gaudi directly on TGI β no more finicking on a custom repository π
15
+
15
16
This integration supports Intel's full line of Gaudi hardware:
16
17
- Gaudi1 π»: Available on [AWS EC2 DL1 instances](https://aws.amazon.com/ec2/instance-types/dl1/)
17
18
- Gaudi2 π»π»: Available on [Intel Dev Cloud](https://ai.cloud.intel.com/)
@@ -34,11 +35,13 @@ The easiest way to run TGI on Gaudi is to use our official Docker image. You nee
34
35
model=meta-llama/Meta-Llama-3.1-8B-Instruct
35
36
volume=$PWD/data # share a volume with the Docker container to avoid downloading weights every run
36
37
hf_token=YOUR_HF_ACCESS_TOKEN
37
-
docker run --runtime=habana --cap-add=sys_nice --ipc=host
For comprehensive documentation on using TGI with Gaudi, including how-to guides and advanced configurations, refer to the new dedicated [Gaudi backend documentation](https://huggingface.co/docs/text-generation-inference/backends/gaudi).
58
+
54
59
## π Top features
55
60
56
61
We have optimized the following models for both single and multi-card configurations. This means these models run as fast as possible on Intel Gaudi. We've specifically optimized the modeling code to target Intel Gaudi hardware, ensuring we offer the best performance and fully utilize Gaudi's capabilities:
@@ -65,10 +70,10 @@ We have optimized the following models for both single and multi-card configurat
65
70
- Gemma (7B)
66
71
- Llava-v1.6-Mistral-7B
67
72
68
-
Furthermore, we also support all models implemented in Transformers library, providing a [fallback mechanism](https://huggingface.co/docs/text-generation-inference/basic_tutorials/non_core_models) that ensures you can still run any model on Gaudi hardware even if it's not yet specifically optimized.
73
+
Furthermore, we also support all models implemented in the [Transformers library](https://huggingface.co/docs/transformers/index), providing a [fallback mechanism](https://huggingface.co/docs/text-generation-inference/basic_tutorials/non_core_models) that ensures you can still run any model on Gaudi hardware even if it's not yet specifically optimized.
69
74
70
75
πββοΈ We also offer many advanced features on Gaudi hardware, such as FP8 quantization thanks to [Intel Neural Compressor (INC)](https://docs.habana.ai/en/latest/PyTorch/Inference_on_PyTorch/Quantization/Inference_Using_FP8.html), enabling even greater performance optimizations.
71
76
72
77
## πͺ Getting Involved
73
78
74
-
We invite the community to try out TGI on Gaudi hardware and provide feedback. The full documentation is available in the TGI repository. π If you're interested in contributing, check out our contribution guidelines or open an issue with your feedback on GitHub. π€ By bringing Intel Gaudi support directly into TGI, we're continuing our mission to provide flexible, efficient, and production-ready tools for deploying LLMs. We're excited to see what you'll build with this new capability! π
79
+
We invite the community to try out TGI on Gaudi hardware and provide feedback. The full documentation is available in the [TGI Gaudi backend documentation](https://huggingface.co/docs/text-generation-inference/backends/gaudi). π If you're interested in contributing, check out our contribution guidelines or open an issue with your feedback on GitHub. π€ By bringing Intel Gaudi support directly into TGI, we're continuing our mission to provide flexible, efficient, and production-ready tools for deploying LLMs. We're excited to see what you'll build with this new capability! π
0 commit comments