Davi Silva df41a57e09 Fix: missing n_kv_heads for smaller models from huggingface (#2438) ...

* fix: missing n_kv_heads for smaller models from huggingface

* a lil golfing

2023-11-25 10:29:04 -08:00

..

mlperf

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

sovits_helpers

Implementation of SoftVC VITS SVC model (#1371)

2023-08-13 19:43:23 -07:00

vgg7_helpers

waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117)

2023-10-19 22:07:15 -07:00

webgpu/stable_diffusion

add name support to fetch (#2407)

2023-11-23 14:16:17 -08:00

__init__.py

failing llama test

2023-03-11 16:28:10 -08:00

beautiful_cartpole.py

A beautiful MNIST training example (#2272)

2023-11-17 19:42:43 -08:00

beautiful_mnist.py

a lil more beautiful_mnist

2023-11-17 19:53:06 -08:00

benchmark_train_efficientnet.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

compile_efficientnet.py

add name support to fetch (#2407)

2023-11-23 14:16:17 -08:00

compile_tensorflow.py

moved extras/jit.py -> tinygrad/jit.py (#599)

2023-02-25 08:32:33 -08:00

efficientnet.py

torch and numpy don't share ops anymore (#2412)

2023-11-23 16:58:10 -08:00

f16_w_uint32.py

add exp2 (#2192)

2023-10-31 17:48:42 -07:00

gpt2.py

multibatch gpt2 (#2432)

2023-11-24 18:10:10 -08:00

handcode_resnet50_opt.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

hlb_cifar10.py

feat: remove cache_id (#2236)

2023-11-08 08:09:21 -08:00

index.html

Enable Multi-Output Export (#2179)

2023-10-30 18:42:26 -07:00

llama.py

Fix: missing n_kv_heads for smaller models from huggingface (#2438)

2023-11-25 10:29:04 -08:00

mask_rcnn.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

mnist_gan.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

serious_mnist.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

simple_conv_bn.py

with Tensor.train() (#1935)

2023-09-28 18:02:31 -07:00

so_vits_svc.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00

stable_diffusion.py

add name support to fetch (#2407)

2023-11-23 14:16:17 -08:00

train_efficientnet.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

train_resnet.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

transformer.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

vgg7.py

waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117)

2023-10-19 22:07:15 -07:00

vit.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

vits.py

Update VITS to use fetch helper (#2422)

2023-11-24 08:50:03 -08:00

whisper.py

Update Whisper to use fetch helper (#2401)

2023-11-23 12:59:59 -08:00

yolov3.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

yolov8-onnx.py

[ready] Replacing os with pathlib (#1708)

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

yolov8.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00