configuration.json

{
    "framework": "pytorch",
    "task": "text-generation",
    "model": {
        "type": "llama2"
    },
    "pipeline": {
        "type": "colossal-llama-2-7b-base-text-generation-pipe"
    }
}