Lorax.Params (lorax v0.1.0)
Helper module for loading, downloading, filtering, and calculating the size of Axon parameters
Summary
Functions
Loads parameters from file path
Returns LoRA only params from a merged param map
Creates a Kino widget for downloading params map. Must be placed in the last line of a Livebook cell.
Creates a Kino widget for uploading serialized params file. Must be placed in the last line of a Livebook cell.
Calculates total bytes of all the tensors inside a parameter map
Functions
file_load!(params_path)
Loads parameters from file path
filter(lora_merged_params, original_params)
Returns LoRA only params from a merged param map
kino_download(params, filename \\ "params.lorax", label \\ "Download Params")
Creates a Kino widget for downloading params map. Must be placed in the last line of a Livebook cell.
kino_file_load!(kino_input)
Creates a Kino widget for uploading serialized params file. Must be placed in the last line of a Livebook cell.
size(params)
Calculates total bytes of all the tensors inside a parameter map