mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-21 04:42:33 +08:00
Vehicle Researcher
db5cbadcf2
date: 2025-06-05T19:54:08 master commit: 8aadf02b2fd91f4e1285e18c2c7feb32d93b66f5
8 lines
258 B
Python
8 lines
258 B
Python
import pathlib
|
|
from tinygrad import Tensor, Device, Context
|
|
|
|
if __name__ == "__main__":
|
|
with Context(DEBUG=2):
|
|
disk_llama = Tensor(pathlib.Path("/raid/weights/LLaMA-3/8B/consolidated.00.pth"))
|
|
device_llama = disk_llama.to(Device.DEFAULT).realize()
|