runtime_env.yaml 222 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
working_dir: ./
excludes: ["/.git/"]
env_vars:
chenych's avatar
v0.3.0  
chenych committed
4
  TOKENIZERS_PARALLELISM: true
chenych's avatar
update  
chenych committed
5
  NCCL_DEBUG: "WARN"
chenych's avatar
v0.3.0  
chenych committed
6
  VLLM_LOGGING_LEVEL: "INFO"
chenych's avatar
chenych committed
7
  TORCH_NCCL_AVOID_RECORD_STREAMS: "1"
chenych's avatar
update  
chenych committed
8
  PYTORCH_CUDA_ALLOC_CONF: "expandable_segments:False"