Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 0faad8c

Browse files
authored
Set PYTORCH_MPS_HIGH_WATERMARK_RATIO in workflow
1 parent d80c64f commit 0faad8c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.github/workflows/pull.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -695,6 +695,7 @@ jobs:
695695
export MODEL_PATH=checkpoints/stories15M/stories15M.pt
696696
export MODEL_NAME=stories15M
697697
export MODEL_DIR=/tmp
698+
export PYTORCH_MPS_HIGH_WATERMARK_RATIO=10.0
698699
699700
python3 torchchat.py generate --device mps --checkpoint-path ${MODEL_PATH} --temperature 0
700701

0 commit comments

Comments
 (0)