GradLLM / .huggingface.yml
Mungert's picture
Update .huggingface.yml
eaca79c verified
raw
history blame
563 Bytes
# .huggingface.yml
# Configures the Space build and runtime environment
sdk: python
python_version: "3.10"
env:
# Force oneDNN to use BF16 math when available
DNNL_DEFAULT_FPMATH_MODE: BF16
# Enable verbose oneDNN kernel logging (see logs at startup/inference)
ONEDNN_VERBOSE: "1"
# Optional: limit MKL threading to avoid oversubscription
OMP_NUM_THREADS: "2"
KMP_AFFINITY: "granularity=fine,compact,1,0"
# Optional: if you want deterministic installs, you can specify a custom pip index
# and requirements
# requirements_file: requirements.txt