fix imports for examples/transformer.py (#1136)

2023-07-05 08:15:13 -07:00
parent c3173ff281
commit d363d25ee2
1 changed files with 2 additions and 1 deletions
@@ -2,7 +2,8 @@
 import numpy as np
 import random

-from tinygrad.nn.optim import Adam, get_parameters
+from tinygrad.state import get_parameters
+from tinygrad.nn.optim import Adam
 from extra.training import train, evaluate
 from models.transformer import Transformer