class

MicroGPT::FutureModel

Inherits Reference < Object

Constructors

new(model : MiniGPT)
Source

Instance methods

causal_mask
Source
future_head
Source
future_mask
Source
generate(start_ids : Array(Int32), max_tokens : Int32, temperature : Float64 = 1.0) : Array(Int32)
Source
model
Source
param_count
Source
train_step(input_ids : Array(Int32), targets : Array(Int32)) : Tuple(Float64, Float64, Float64)
Source