Skip to content

Commit

Permalink
Update dependencies in notebooks
Browse files Browse the repository at this point in the history
  • Loading branch information
jonatanklosko committed Feb 23, 2024
1 parent 7b0d06c commit 9c55e14
Show file tree
Hide file tree
Showing 5 changed files with 19 additions and 25 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ First add Bumblebee and EXLA as dependencies in your `mix.exs`. EXLA is an optio
```elixir
def deps do
[
{:bumblebee, "~> 0.4.2"},
{:bumblebee, "~> 0.5.0"},
{:exla, ">= 0.0.0"}
]
end
Expand All @@ -47,7 +47,7 @@ In notebooks and scripts, use the following `Mix.install/2` call to both install
```elixir
Mix.install(
[
{:bumblebee, "~> 0.4.2"},
{:bumblebee, "~> 0.5.0"},
{:exla, ">= 0.0.0"}
],
config: [nx: [default_backend: EXLA.Backend]]
Expand Down
10 changes: 5 additions & 5 deletions notebooks/examples.livemd
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@

```elixir
Mix.install([
{:bumblebee, "~> 0.4.2"},
{:nx, "~> 0.6.1"},
{:exla, "~> 0.6.1"},
{:axon, "~> 0.6.0"},
{:kino, "~> 0.10.0"}
{:bumblebee, "~> 0.5.0"},
{:nx, "~> 0.7.0"},
{:exla, "~> 0.7.0"},
{:axon, "~> 0.6.1"},
{:kino, "~> 0.12.0"}
])

Nx.global_default_backend(EXLA.Backend)
Expand Down
8 changes: 4 additions & 4 deletions notebooks/fine_tuning.livemd
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@

```elixir
Mix.install([
{:bumblebee, "~> 0.4.2"},
{:axon, "~> 0.6.0"},
{:nx, "~> 0.6.1"},
{:exla, "~> 0.6.1"},
{:bumblebee, "~> 0.5.0"},
{:nx, "~> 0.7.0"},
{:exla, "~> 0.7.0"},
{:axon, "~> 0.6.1"},
{:explorer, "~> 0.7.0"}
])

Expand Down
11 changes: 4 additions & 7 deletions notebooks/llama.livemd
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,10 @@

```elixir
Mix.install([
# {:bumblebee, "~> 0.4.2"},
# {:nx, "~> 0.6.1"},
# {:exla, "~> 0.6.1"},
{:bumblebee, github: "elixir-nx/bumblebee"},
{:nx, github: "elixir-nx/nx", sparse: "nx", override: true},
{:exla, github: "elixir-nx/nx", sparse: "exla", override: true},
{:kino, "~> 0.11.0"}
{:bumblebee, "~> 0.5.0"},
{:nx, "~> 0.7.0"},
{:exla, "~> 0.7.0"},
{:kino, "~> 0.12.0"}
])

Nx.global_default_backend({EXLA.Backend, client: :host})
Expand Down
11 changes: 4 additions & 7 deletions notebooks/stable_diffusion.livemd
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,10 @@

```elixir
Mix.install([
# {:bumblebee, "~> 0.4.2"},
# {:nx, "~> 0.6.1"},
# {:exla, "~> 0.6.1"},
{:bumblebee, github: "elixir-nx/bumblebee"},
{:nx, github: "elixir-nx/nx", sparse: "nx", override: true},
{:exla, github: "elixir-nx/nx", sparse: "exla", override: true},
{:kino, "~> 0.11.0"}
{:bumblebee, "~> 0.5.0"},
{:nx, "~> 0.7.0"},
{:exla, "~> 0.7.0"},
{:kino, "~> 0.12.0"}
])

Nx.global_default_backend({EXLA.Backend, client: :host})
Expand Down

0 comments on commit 9c55e14

Please sign in to comment.