Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Select an option

  • Save gurusura/4c2a2dec8fcbefdcc0c2c27bacc71d9f to your computer and use it in GitHub Desktop.

Select an option

Save gurusura/4c2a2dec8fcbefdcc0c2c27bacc71d9f to your computer and use it in GitHub Desktop.

Revisions

  1. @YouJiacheng YouJiacheng revised this gist Jan 14, 2025. 1 changed file with 32 additions and 0 deletions.
    32 changes: 32 additions & 0 deletions pyproject.toml
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,32 @@
    [project]
    name = "modded-nanogpt"
    version = "0.1.0"
    description = "Add your description here"
    readme = "README.md"
    requires-python = "==3.12.*"
    dependencies = [
    "numpy>=2.1.3",
    "torch",
    "pytorch-triton>=3.2.0",
    "huggingface-hub>=0.26.2",
    "tqdm>=4.67.0",
    "pip>=24.3.1",
    ]

    [tool.uv]
    environments = [
    "sys_platform == 'linux'",
    ]

    [tool.uv.sources]
    torch = [
    { url = "https://download.pytorch.org/whl/nightly/cu126/torch-2.7.0.dev20250110%2Bcu126-cp312-cp312-manylinux_2_28_x86_64.whl" },
    ]
    pytorch-triton = [
    { index = "pytorch-nightly-cu126" },
    ]

    [[tool.uv.index]]
    name = "pytorch-nightly-cu126"
    url = "https://download.pytorch.org/whl/nightly/cu126"
    explicit = true
  2. @YouJiacheng YouJiacheng created this gist Jan 13, 2025.
    2,182 changes: 2,182 additions & 0 deletions bfcfc9e4-61d0-4bc3-8d66-f9185bedff83.txt
    2,182 additions, 0 deletions not shown because the diff is too large. Please use a local Git client to view these changes.