github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

move device to device.py (#2466)

Browse Source 
* move device to device.py

* pylint test --disable R,C,W,E --enable E0611

* fix tests
This commit is contained in:
George Hotz
2023-11-27 11:34:37 -08:00
committed by GitHub
parent 262cd26d28
commit 9e07824542
62 changed files with 281 additions and 264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/webgpu/stable_diffusion/compile.py
View File

@@ -3,7 +3,7 @@ from extra.export_model import compile_net, jit_model
from examples.stable_diffusion import StableDiffusion
from tinygrad.nn.state import get_state_dict, safe_save, safe_load_metadata, torch_load, load_state_dict
from tinygrad.tensor import Tensor
from tinygrad.ops import Device
from tinygrad import Device
from tinygrad.helpers import fetch
from typing import NamedTuple, Any, List
from pathlib import Path