github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

move globalcounters to ops (#2960)

Browse Source 
* move globalcounters to ops

* missed a few

* sick of that failing
This commit is contained in:
George Hotz
2024-01-01 14:21:02 -08:00
committed by GitHub
parent 8291986959
commit c81ce9643d
23 changed files with 52 additions and 282 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/stable_diffusion.py
View File

@@ -10,8 +10,8 @@ from PIL import Image
import numpy as np
from tqdm import tqdm
from tinygrad.tensor import Tensor
from tinygrad import Device
from tinygrad.helpers import dtypes, GlobalCounters, Timing, Context, getenv, fetch, colored
from tinygrad import Device, GlobalCounters
from tinygrad.helpers import dtypes, Timing, Context, getenv, fetch, colored
from tinygrad.nn import Conv2d, Linear, GroupNorm, LayerNorm, Embedding
from tinygrad.nn.state import torch_load, load_state_dict, get_state_dict
from tinygrad.jit import TinyJit