import Generation
Coverage
Declarations
Interfaces
protocol Generation
struct GenerationConfig
Essentials taken from https://github.com/huggingface/transformers/blob/main/src/transformers/generation/configuration_utils.py
enum GenerationMode
typealias GenerationOutput = [Int]
typealias InputTokens = [Int]
typealias NextTokenModel = (InputTokens, GenerationConfig) -> any MLShapedArrayProtocol
A callable (a model, usually), that predicts the next token after a given sequence
typealias PredictionStringCallback = (String) -> Void
typealias PredictionTokensCallback = (GenerationOutput) -> Void
import Hub
import HubCLI
import Models
import TensorUtils
import Tokenizers
import TransformersCLI