Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@murrellb
murrellb authored Nov 30, 2024
1 parent 687192c commit 5a10927
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,7 @@ In addition to `JSON3` and `Jjama3`, ensure `CUDA.jl` and `Flux.jl` are installe
using CUDA, Flux, JSON3, Jjama3
```

You might also need to install `cuDNN.jl` and run `using cuDNN` on some systems.
You might also need to install `cuDNN` and run `using cuDNN` on some systems.

For sampling, you can pass `device = gpu` to the `generate` function:

Expand Down

0 comments on commit 5a10927

Please sign in to comment.