From da694d42416b090f4bfa9c0e05a5b544dccd09ca Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 22 Aug 2023 21:30:55 -0700 Subject: [PATCH] move that image import --- tinygrad/tensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 0267a43ed7..71d7b27075 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -706,8 +706,8 @@ for device in Device._buffers: setattr(Tensor, f"{device.lower()}", partialmethod(Tensor.to, device)) setattr(Tensor, f"{device.lower()}_", partialmethod(Tensor.to_, device)) -# if IMAGE>0 we install these replacement functions in Tensor (hack!) -from tinygrad.nn.image import image_conv2d, image_dot if IMAGE: + # if IMAGE>0 we install these replacement functions in Tensor (hack!) + from tinygrad.nn.image import image_conv2d, image_dot setattr(Tensor, "conv2d", image_conv2d) setattr(Tensor, "dot", image_dot)