From d363d25ee212cfac1e5bf91c1802f5a22274958a Mon Sep 17 00:00:00 2001
From: nimlgen <138685161+nimlgen@users.noreply.github.com>
Date: Wed, 5 Jul 2023 18:15:13 +0300
Subject: [PATCH] fix imports for examples/transformer.py (#1136)

---
 examples/transformer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/transformer.py b/examples/transformer.py
index 855dfa20fe..7bc2157e9f 100755
--- a/examples/transformer.py
+++ b/examples/transformer.py
@@ -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