optimum-cli export neuron  --model 01-ai/Yi-1.5-6B-Chat --batch_size 1 --sequence_length 1024 --num_cores 2 --auto_cast_type fp16  ./models/01-ai/Yi-1.5-6B-Chat

Downloads last month: 16

Inference Examples

Text Generation

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.