logo

InferX AI Function Platform (Lambda Function for Inference)

    --   Serve tens models in one box with ultra-fast (<2 sec) cold start (contact: support@inferx.net)



Model gpt2-xl

namespace model name standby gpu standby pageable standby pinned memory gpu count vRam (MB) cpu memory (MB) state revision
openai-community gpt2-xl Blob Blob Blob 1 12000 12.0 18000 Normal 237

Image

Prompt



Sample Rest Call

Pods

tenant namespace pod name state require resource allocated resource
public openai-community public/openai-community/gpt2-xl/237/958 Standby {'CPU': 12000, 'Mem': 18000, 'GPU': {'Type': 'Any', 'Count': 1, 'vRam': 12000}} {'nodename': 'node3', 'CPU': 12000, 'Mem': 18000, 'GPUType': 'A4000', 'GPUs': {'vRam': 0, 'map': {}, 'slotSize': 0, 'totalSlotCnt': 0}, 'MaxContextPerGPU': 2}

Func

{
 "image": "vllm/vllm-openai:v0.6.2",
 "commands": [
  "--model",
  "openai-community/gpt2-xl",
  "--disable-custom-all-reduce",
  "--enforce-eager",
  "--max-model-len",
  "800"
 ],
 "envs": [
  [
   "LD_LIBRARY_PATH",
   "/usr/local/lib/python3.12/dist-packages/nvidia/cuda_nvrtc/lib/:$LD_LIBRARY_PATH"
  ]
 ],
 "mounts": [
  {
   "hostpath": "/home/brad/cache",
   "mountpath": "/root/.cache/huggingface"
  }
 ],
 "endpoint": {
  "port": 8000,
  "schema": "Http",
  "probe": "/health"
 },
 "version": 237,
 "entrypoint": [],
 "resources": {
  "CPU": 12000,
  "Mem": 18000,
  "GPU": {
   "Type": "Any",
   "Count": 1,
   "vRam": 12000
  }
 },
 "standby": {
  "gpu": "Blob",
  "pageable": "Blob",
  "pinned": "Blob"
 },
 "probe": {
  "port": 80,
  "schema": "Http",
  "probe": "/health"
 },
 "sample_query": {
  "apiType": "openai",
  "path": "v1/completions",
  "prompt": "Here is a recipe for vegan banana bread:",
  "body": {
   "max_tokens": "600",
   "model": "openai-community/gpt2-xl",
   "stream": "true",
   "temperature": "0"
  }
 }
}