environment.json

{
  "packages": {
    "torch": "2.8.0",
    "transformers": "5.17.0",
    "safetensors": "0.8.0",
    "pydantic": "2.13.5",
    "peft": "0.21.0",
    "triton": "3.7.1",
    "flash-linear-attention": "0.5.2",
    "tokenizers": "0.23.2",
    "huggingface-hub": "1.32.0",
    "einops": "0.8.2"
  },
  "python": "3.12.10 (main, Apr  9 2025, 08:55:05) [GCC 11.4.0]",
  "cuda": "12.8",
  "gpu": "NVIDIA H20"
}