mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
don't fail when termcolor is not installed (#436)
This commit is contained in:
@@ -11,9 +11,11 @@ from tinygrad.tensor import Tensor
|
||||
from tinygrad.nn import Conv2d
|
||||
try:
|
||||
from termcolor import colored
|
||||
except ImportError:
|
||||
colored = lambda x, _: x
|
||||
try:
|
||||
from tinygrad.llops.ops_gpu import CL
|
||||
except ImportError:
|
||||
colored = None
|
||||
CL = None
|
||||
|
||||
IN_CHANS = [int(x) for x in os.getenv("IN_CHANS", "4,16,64").split(",")]
|
||||
|
||||
Reference in New Issue
Block a user