bump line count to 14500

This commit is contained in:
George Hotz
2025-06-23 15:29:03 -07:00
parent 79d7cdd9ba
commit ae4d2d71b4
2 changed files with 2 additions and 3 deletions

View File

@@ -371,8 +371,8 @@ jobs:
run: PYTHONPATH="." python test/external/external_uop_gc.py
- name: Run process replay tests
uses: ./.github/actions/process-replay
- name: Repo line count < 14000 lines
run: MAX_LINE_COUNT=14000 python sz.py
- name: Repo line count < 14500 lines
run: MAX_LINE_COUNT=14500 python sz.py
fuzzing:
name: Fuzzing

View File

@@ -12,7 +12,6 @@ if __name__ == "__main__":
with Timing("total "):
with Timing("***** create model in "):
# NOTE: max_context=None means no kv cache. kv cache has realize in the model
model = Transformer(**MODEL_PARAMS[model_size]["args"], linear=nn.Linear, embedding=nn.Embedding,
max_context=1024, jit=True, disable_kv_cache=True)