File size: 1,268 Bytes
c12c877
 
 
259b220
 
c289686
259b220
 
 
 
 
82e11c0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
---
license: apache-2.0
---
# AWS Neuron optimum model cache

This repository contains cached neuron compilation artifacts for the most popular models on the Hugging Face Hub.

## Inference

### LLM models 

The transparent caching mechanism included in `optimum-neuron` and `NeuronX TGI`, makes it easier to export and deploy cached models to Neuron platforms such as Trainium and Inferentia.

To deploy directly any cached model to SageMaker:
- go to the model page,
- select "Deploy" in the top right corner,
- select "AWS SageMaker" in the drop-down,
- select the "AWS Inferentia & Trainium" tab,
- copy the code snippet.

You can now paste the code snippet in your deployment script or notebook, following the instructions in the comment.

To export a model to Neuron and save it locally, please follow the instructions in the `optimum-neuron` [documentation](https://huggingface.co./docs/optimum-neuron/guides/export_model).

For a list of the cached models and configurations, please refer to the inference cache [configuration files](https://huggingface.co./aws-neuron/optimum-neuron-cache/tree/main/inference-cache-config).

Alternatively, you can use the `optimum-cli neuron cache lookup` command to look for a specific model and see the cached configurations.