Fix docs warning

elixir-nx · Oct 7, 2024 · f17779f · f17779f
1 parent a77617d
commit f17779f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -30,8 +30,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 * Whisper serving to stream input using ffmpeg when given a file path ([#361](https://github.com/elixir-nx/bumblebee/pull/361))
 * **(Breaking)** Changed model params to use `%Axon.ModelState{}`, matching Axon v0.7.0 ([#375](https://github.com/elixir-nx/bumblebee/pull/375))
 
-### Removed
-
 ### Fixed
 
 * Token classification crashing when all tokens are special

diff --git a/lib/bumblebee/text/llama.ex b/lib/bumblebee/text/llama.ex
@@ -30,7 +30,7 @@ defmodule Bumblebee.Text.Llama do
         default: nil,
         doc: """
         the size of the key, value, and query projection per attention head.
-        Defaults to `div(hidden_size, num_attention_heads)
+        Defaults to `div(hidden_size, num_attention_heads)`
         """
       ],
       num_blocks: [

diff --git a/mix.exs b/mix.exs
@@ -108,6 +108,7 @@ defmodule Bumblebee.MixProject do
           Bumblebee.Vision.Deit,
           Bumblebee.Vision.DinoV2,
           Bumblebee.Vision.ResNet,
+          Bumblebee.Vision.Swin,
           Bumblebee.Vision.Vit
         ],
         Preprocessors: [
@@ -138,7 +139,7 @@ defmodule Bumblebee.MixProject do
           Bumblebee.Text.WhisperGenerationConfig
         ]
       ],
-      groups_for_functions: [
+      groups_for_docs: [
         # Bumblebee
         Models: &(&1[:type] == :model),
         Featurizers: &(&1[:type] == :featurizer),