Bump versions (#427)

Author: jonatanklosko

.github/workflows/nightly.yml

@@ -4,8 +4,8 @@ on:
     - cron: "0 0 * * *"
   workflow_dispatch:
 env:
-  elixir: 1.14.0
-  otp: 24.0
+  elixir: "1.19.2"
+  otp: "28.1.1"
 jobs:
   test:
     name: Test all
@@ -15,12 +15,12 @@ jobs:
       XLA_CACHE_DIR: ${{ github.workspace }}/cache/xla
       LIBTORCH_DIR: ${{ github.workspace }}/cache/libtorch
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v5
       - uses: erlef/setup-beam@v1
         with:
           otp-version: ${{env.otp}}
           elixir-version: ${{env.elixir}}
-      - uses: actions/cache@v3
+      - uses: actions/cache@v4
         with:
           path: |
             deps

.github/workflows/test.yaml

@@ -14,27 +14,27 @@ jobs:
       matrix:
         include:
           - pair:
-              elixir: "1.15.4"
-              otp: "26.0.2"
+              elixir: "1.19.2"
+              otp: "28.1.1"
             lint: true
             slow: true
           - pair:
-              elixir: "1.14.5"
-              otp: "25.3.2.2"
+              elixir: "1.15.8"
+              otp: "26.2.5"
     env:
       MIX_ENV: test
       XLA_CACHE_DIR: ${{ github.workspace }}/cache/xla
       LIBTORCH_DIR: ${{ github.workspace }}/cache/torch
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v5
         with:
           # We need the previous commit for git diff later
           fetch-depth: 2
       - uses: erlef/setup-beam@v1
         with:
           otp-version: ${{ matrix.pair.otp }}
           elixir-version: ${{ matrix.pair.elixir }}
-      - uses: actions/cache@v3
+      - uses: actions/cache@v4
         with:
           path: |
             deps

lib/bumblebee/conversion/pytorch_params.ex

@@ -84,12 +84,17 @@ defmodule Bumblebee.Conversion.PyTorchParams do
 
   defp state_dict?(%{} = dict) when not is_struct(dict) do
     Enum.all?(dict, fn {key, value} ->
-      is_binary(key) and Nx.LazyContainer.impl_for(value) != nil
+      is_binary(key) and implements_lazy_container?(value)
     end)
   end
 
   defp state_dict?(_other), do: false
 
+  defp implements_lazy_container?(value) do
+    Nx.LazyContainer.impl_for(value) != Nx.LazyContainer.Any or
+      Nx.Container.impl_for(value) != nil
+  end
+
   defp init_params(model, params_expr, pytorch_state, params_mapping) do
     layers =
       model

mix.exs

@@ -10,7 +10,7 @@ defmodule Bumblebee.MixProject do
       version: @version,
       description: @description,
       name: "Bumblebee",
-      elixir: "~> 1.14",
+      elixir: "~> 1.15",
       elixirc_paths: elixirc_paths(Mix.env()),
       start_permanent: Mix.env() == :prod,
       deps: deps(),
@@ -44,7 +44,7 @@ defmodule Bumblebee.MixProject do
       {:unpickler, "~> 0.1.0"},
       {:safetensors, "~> 0.1.3"},
       {:jason, "~> 1.4.0"},
-      {:unzip, "~> 0.12.0"},
+      {:unzip, "~> 0.12.0 or ~> 0.13.0"},
       {:progress_bar, "~> 3.0"},
       {:stb_image, "~> 0.6.0", only: :test},
       {:bypass, "~> 2.1", only: :test},