We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cc2bbb0 commit ba97fd0Copy full SHA for ba97fd0
test/external/external_benchmark_disk_raw.py
@@ -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