Yixiang Gao 66a6bbd029 codellama (#1702) ...

* add codellama with pre-downloaded weights

* add rope_theta, fix param

* fix test

* add 7B-Python

* add 7B-Instruct

* replace single quotes with doulbe

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>

2023-09-02 08:45:12 -07:00

..

external

codellama (#1702)

2023-09-02 08:45:12 -07:00

extra

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

models

faster gpt2 jit path and gpt2 in test_real_world (#1738)

2023-09-02 08:39:12 -07:00

unit

visitor pattern for sym_infer and unit tests (#1733)

2023-09-01 09:47:45 -07:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

Docker fix (#1039)

2023-06-25 10:38:58 -07:00

test_allocators.py

allow zerosized tensors (#1659)

2023-08-30 10:39:24 -07:00

test_assign.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_cache_collector.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv_shapetracker.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv.py

Uast start (#1650)

2023-08-23 12:00:06 -07:00

test_custom_function.py

Optional: Reduce line count and simplify the LazyBuffer interface (#1642)

2023-08-22 21:01:10 -07:00

test_dtype.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_helpers.py

add dtypes test (#1682)

2023-08-28 08:12:15 -07:00

test_jit.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_lazybuffer.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_linearizer.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_nn.py

sparse cat cross entropy (#1597)

2023-08-21 14:14:54 -07:00

test_ops.py

remove Token class (#1723)

2023-09-01 12:53:07 -07:00

test_optim.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_randomness.py

Tensor.uniform with dtype=int bug fix (#1593)

2023-08-26 01:59:53 -04:00

test_specific_conv.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_speed_v_torch.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_symbolic_jit.py

llama JIT python runtime speedup (#1633)

2023-08-30 07:51:05 -07:00

test_symbolic_ops.py

fix symbolic_ops tests with Tensor.training=True (#1686)

2023-08-26 23:19:56 -04:00

test_symbolic_shapetracker.py

support symbols in shrink (#1611)

2023-08-22 09:08:21 -07:00

test_tensor.py

allow zerosized tensors (#1659)

2023-08-30 10:39:24 -07:00

test_uops.py

render const (#1736)

2023-09-01 19:01:43 -07:00

test_webgpu.js

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00