Skip to content

Commit

Permalink
Up
Browse files Browse the repository at this point in the history
  • Loading branch information
jonatanklosko committed May 30, 2024
1 parent 3a808c0 commit f89f96a
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions lib/bumblebee/text/phi3.ex
Original file line number Diff line number Diff line change
Expand Up @@ -89,21 +89,21 @@ defmodule Bumblebee.Text.Phi3 do
Shared.token_options(pad_token_id: 32000)

@moduledoc """
Phi model family.
Phi-3 model family.
## Architectures
* `:base` - plain Phi without any head on top
* `:base` - plain Phi-3 without any head on top
* `:for_causal_language_modeling` - Phi with a language modeling
* `:for_causal_language_modeling` - Phi-3 with a language modeling
head. The head returns logits for each token in the original
sequence
* `:for_sequence_classification` - Phi with a sequence
* `:for_sequence_classification` - Phi-3 with a sequence
classification head. The head returns logits corresponding to
possible classes
* `:for_token_classification` - Phi with a token classification
* `:for_token_classification` - Phi-3 with a token classification
head. The head returns logits for each token in the original
sequence
Expand Down

0 comments on commit f89f96a

Please sign in to comment.