Skip to content

Commit 1531aa5

Browse files
authored
Merge pull request #7 from arthurdouillard/patch-1
Fix code example in README
2 parents e8ff410 + 9659a54 commit 1531aa5

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

README.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -255,10 +255,9 @@ Specifically, if you want to use OpenDiLoCo in your own training script, you can
255255
from functools import partial
256256

257257
from open_diloco.hivemind_diloco import DiLoCoOptimizer
258-
259-
import
260258
from hivemind.dht.dht import DHT
261259

260+
262261
dht = DHT(start=True, initial_peers=os.environ["PEERS"])
263262

264263
inner_optimizer = partial(torch.optim.AdamW, lr=4e-4) # optimizer need to be function

0 commit comments

Comments
 (0)