mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-07 22:23:55 -05:00
bump line count to 14500
This commit is contained in:
4
.github/workflows/test.yml
vendored
4
.github/workflows/test.yml
vendored
@@ -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
|
||||
|
||||
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user