mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-05 21:42:05 +08:00
firestar5683
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()
|