mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-30 19:33:45 +08:00
github-actions[bot]
eefc084302
version: sunnypilot v2025.003.000 (dev) date: 2025-12-18T05:48:43 master commit: 16c052af0835f049a67b80251d8cc1114fc08bb4
13 lines
346 B
Python
13 lines
346 B
Python
#!/usr/bin/env python3
|
|
from tinygrad import Tensor, TinyJit, nn
|
|
from extra.models.llama import FeedForward
|
|
|
|
if __name__ == "__main__":
|
|
model = FeedForward(4096, 14336)
|
|
for x in nn.state.get_parameters(model): x.replace(x.half()).realize()
|
|
jrun = TinyJit(model)
|
|
for i in range(5):
|
|
print(f"*** run {i}")
|
|
jrun(Tensor.rand(1, 4096))
|
|
|