Skip to content

Commit 82a3bad

Browse files
Update README.md (#41)
Co-authored-by: Romain Huet <[email protected]>
1 parent f409636 commit 82a3bad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ Check out our [awesome list](./awesome-gpt-oss.md) for a broader collection of g
116116
This repository provides a collection of reference implementations:
117117

118118
- **Inference:**
119-
- [`torch`](#reference-pytorch-implementation) — a non-optimized [PyTorch](https://pytorch.org/) implementation for educational purposes only. Requires at least 4x H100s because it's not optimized
119+
- [`torch`](#reference-pytorch-implementation) — a non-optimized [PyTorch](https://pytorch.org/) implementation for educational purposes only. Requires at least 4× H100 GPUs due to lack of optimization.
120120
- [`triton`](#reference-triton-implementation-single-gpu) — a more optimized implementation using [PyTorch](https://pytorch.org/) & [Triton](https://github.com/triton-lang/triton) incl. using CUDA graphs and basic caching
121121
- [`metal`](#reference-metal-implementation) — a Metal-specific implementation for running the models on Apple Silicon hardware
122122
- **Tools:**
@@ -227,7 +227,7 @@ To perform inference you'll need to first convert the SafeTensor weights from Hu
227227
python gpt_oss/metal/scripts/create-local-model.py -s <model_dir> -d <output_file>
228228
```
229229

230-
Or download the pre-converted weight:
230+
Or download the pre-converted weights:
231231

232232
```shell
233233
hf download openai/gpt-oss-120b --include "metal/*" --local-dir gpt-oss-120b/metal/

0 commit comments

Comments
 (0)