浏览代码

Okay let's turn no_grad back on. We'll worry about that when tinygrad training works

Nel Nibcord 8 月之前
父节点
当前提交
bcf87e79b7
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      exo/inference/tinygrad/inference.py

+ 1 - 1
exo/inference/tinygrad/inference.py

@@ -16,7 +16,7 @@ from .losses import length_masked_ce_loss
 from collections import OrderedDict
 from collections import OrderedDict
 import asyncio
 import asyncio
 
 
-Tensor.no_grad = False
+Tensor.no_grad = True 
 # default settings
 # default settings
 TEMPERATURE = int(os.getenv("TEMPERATURE", 0.85))
 TEMPERATURE = int(os.getenv("TEMPERATURE", 0.85))
 TOP_K = 25
 TOP_K = 25