github-actions[bot]
3dcc048299
date: 2026-06-19T21:43:27 master commit: 5d90689776fdc7a3be31fc1335003aee20a2ba62
9 lines
314 B
Python
9 lines
314 B
Python
import pathlib
|
|
from tinygrad import Tensor, Device, Context
|
|
from tinygrad.helpers import getenv
|
|
|
|
if __name__ == "__main__":
|
|
with Context(DEBUG=2):
|
|
disk_llama = Tensor(pathlib.Path(getenv("TESTFILE", "/raid/weights/LLaMA-3/8B/consolidated.00.pth")))
|
|
device_llama = disk_llama.to(Device.DEFAULT).realize()
|