minor: model.to(device)

[d-kleine]

Bug description

I have noticed a small thing in the code in the "Introduction to PyTorch" section:

LLMs-from-scratch/appendix-A/01_main-chapter-code/code-part2.ipynb

Line 334 in 08040f0

"model = model.to(device) # NEW\n",

Code reference in the print/ebook on p. 281

The assignment is actually not needed as for models (different for tensors), the assignment will be performed in-place and should be slightly faster:

model.to(device)

Idk if this was done here on purpose because at some other code sections of the book this is correctly pointed out, for instance:

LLMs-from-scratch/appendix-E/01_main-chapter-code/appendix-E.ipynb

Line 557 in c278745

"model.to(device); # no assignment model = model.to(device) necessary for nn.Module classes"

What operating system are you using?

None

Where do you run your code?

None

Environment