From f17779fa37bc28f2e8f84f9b4103594cc95f6d4b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jonatan=20K=C5=82osko?= Date: Tue, 8 Oct 2024 01:06:43 +0800 Subject: [PATCH] Fix docs warning --- CHANGELOG.md | 2 -- lib/bumblebee/text/llama.ex | 2 +- mix.exs | 3 ++- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index c07ef736..ec3fd291 100644 --- 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 index 3fed0524..b06a0315 100644 --- 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 index 0d5a669f..bbe4aa3b 100644 --- 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),