Skip to content

Commit 7ba69ff

Browse files
authored
Update README.md (#58)
1 parent 1a9e106 commit 7ba69ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ hf download openai/gpt-oss-20b --include "original/*" --local-dir gpt-oss-20b/
171171

172172
We include an inefficient reference PyTorch implementation in [gpt_oss/torch/model.py](gpt_oss/torch/model.py). This code uses basic PyTorch operators to show the exact model architecture, with a small addition of supporting tensor parallelism in MoE so that the larger model can run with this code (e.g., on 4xH100 or 2xH200). In this implementation, we upcast all weights to BF16 and run the model in BF16.
173173

174-
To run the reference implementation. Install dependencies:
174+
To run the reference implementation, install these dependencies:
175175

176176
```shell
177177
pip install -e ".[torch]"

0 commit comments

Comments
 (0)