Skip to content

Commit ba97fd0

Browse files
committed
hotfix: add test/external/external_benchmark_disk_raw
1 parent cc2bbb0 commit ba97fd0

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
import pathlib
2+
from tinygrad import Tensor, Device, Context
3+
4+
if __name__ == "__main__":
5+
with Context(DEBUG=2):
6+
disk_llama = Tensor(pathlib.Path("/raid/weights/LLaMA-3/8B/consolidated.00.pth"))
7+
device_llama = disk_llama.to(Device.DEFAULT).realize()

0 commit comments

Comments
 (0)