mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
This reverts commit da2efecbe2.
This commit is contained in:
@@ -9,7 +9,7 @@ if __name__ == "__main__":
|
||||
model.load_from_pretrained()
|
||||
run, special_names = jit_model(model, Tensor.randn(1,3,224,224))
|
||||
functions, statements, bufs, _bufs_to_save = compile_net(run, special_names)
|
||||
|
||||
|
||||
state = get_state_dict(model)
|
||||
weights = {id(x.lazydata.realized): name for name, x in state.items()}
|
||||
safe_save(state, path.join(path.dirname(__file__), "net.safetensors"))
|
||||
@@ -28,7 +28,7 @@ if __name__ == "__main__":
|
||||
const getTensorBuffer = (safetensorBuffer, tensorMetadata) => {{
|
||||
return safetensorBuffer.subarray(...tensorMetadata.data_offsets);
|
||||
}}
|
||||
|
||||
|
||||
const createEmptyBuf = (device, size) => {{
|
||||
return device.createBuffer({{size, usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST }});
|
||||
}};
|
||||
@@ -50,7 +50,7 @@ const addComputePass = (device, commandEncoder, pipeline, bufs, workgroup) => {{
|
||||
}};
|
||||
|
||||
{kernel_code}
|
||||
|
||||
|
||||
const setupNet = async (device, safetensor) => {{
|
||||
const metadata = getTensorMetadata(safetensor);
|
||||
|
||||
|
||||
@@ -56,7 +56,7 @@ canvas { display: none; }
|
||||
console.log(`${delta} ms ${label}`);
|
||||
document.getElementById('time').innerHTML = `${delta} ms ${label}`;
|
||||
return out;
|
||||
}
|
||||
}
|
||||
|
||||
const getLabels = async () => (await fetch("https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json")).json();
|
||||
|
||||
@@ -80,7 +80,7 @@ canvas { display: none; }
|
||||
const response = await fetch(resource)
|
||||
if (!response.ok) error("sir. that is not a good URL. try a new one");
|
||||
document.getElementById("imagebox").src = resource
|
||||
|
||||
|
||||
const img = new Image();
|
||||
img.crossOrigin = "Anonymous";
|
||||
img.onload = () => {
|
||||
|
||||
Reference in New Issue
Block a user