Преглед на файлове

bring back TINYGRAD_DEBUG. not sure why it was removed

Alex Cheema преди 1 година
родител
ревизия
980d5d2c5b
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 2 0
      exo/helpers.py

+ 2 - 0
exo/helpers.py

@@ -36,6 +36,8 @@ def get_inference_engine(inference_engine_name):
     return MLXDynamicShardInferenceEngine()
   elif inference_engine_name == "tinygrad":
     from exo.inference.tinygrad.inference import TinygradDynamicShardInferenceEngine
+    import tinygrad.helpers
+    tinygrad.helpers.DEBUG.value = int(os.getenv("TINYGRAD_DEBUG", default="0"))
 
     return TinygradDynamicShardInferenceEngine()
   else: