codellama--CodeLlama-34b-Instruct-hf.yaml

deployment_config:
  autoscaling_config:
    min_replicas: 1
    initial_replicas: 1
    max_replicas: 2
    target_num_ongoing_requests_per_replica: 32
    metrics_interval_s: 10.0
    look_back_period_s: 30.0
    smoothing_factor: 0.6
    downscale_delay_s: 300.0
    upscale_delay_s: 10.0
  max_concurrent_queries: 128
  ray_actor_options:
    resources:
      accelerator_type_a100_80g: 0.01
engine_config:
  model_id: codellama/CodeLlama-34b-Instruct-hf
  hf_model_id: codellama/CodeLlama-34b-Instruct-hf
  type: VLLMEngine
  engine_kwargs:
    trust_remote_code: true
    max_num_batched_tokens: 65536
    max_num_seqs: 128
    gpu_memory_utilization: 0.95
  max_total_tokens: 16384
  generation:
    prompt_format:
      system: "<<SYS>>\n{instruction}\n<</SYS>>\n\n"
      assistant: " {instruction} </s><s>"
      trailing_assistant: ""
      user: "[INST] {system}{instruction} [/INST]"
      system_in_user: true
      default_system_message: ""
    stopping_sequences: []
scaling_config:
  num_workers: 2
  num_gpus_per_worker: 1
  num_cpus_per_worker: 8
  placement_strategy: "STRICT_PACK"
  resources_per_worker:
    accelerator_type_a100_80g: 0.01