Packages

tinkex

0.1.1
0.4.0 0.3.4 0.3.3 0.3.2 0.3.1 0.3.0 0.2.2 0.2.1 0.2.0 0.1.20 0.1.19 0.1.18 0.1.17 0.1.16 0.1.15 0.1.14 0.1.13 0.1.12 0.1.11 0.1.10 0.1.9 0.1.8 0.1.7 0.1.6 0.1.5 0.1.4 0.1.3 0.1.2 0.1.1 0.1.0

Elixir SDK for Tinker: LoRA training, sampling, and future-based workflows with telemetry and HTTP/2.

HexDocs

Current section

Files

Jump to
Readme
30 Versions
5 Dependencies
Dependants
Files
Activity
tinkex lib tinkex api weights.ex
Raw

lib/tinkex/api/weights.ex 

defmodule Tinkex.API.Weights do

  @moduledoc """

  Weight management endpoints.



  Uses :training pool for weight operations.

  """



  @doc """

  Save model weights.

  """

  @spec save_weights(map(), keyword()) ::

          {:ok, map()} | {:error, Tinkex.Error.t()}

  def save_weights(request, opts) do

    Tinkex.API.post(

      "/api/v1/save_weights",

      request,

      Keyword.put(opts, :pool_type, :training)

    )

  end



  @doc """

  Load model weights.

  """

  @spec load_weights(map(), keyword()) ::

          {:ok, map()} | {:error, Tinkex.Error.t()}

  def load_weights(request, opts) do

    Tinkex.API.post(

      "/api/v1/load_weights",

      request,

      Keyword.put(opts, :pool_type, :training)

    )

  end



  @doc """

  Save weights for sampler.

  """

  @spec save_weights_for_sampler(map(), keyword()) ::

          {:ok, map()} | {:error, Tinkex.Error.t()}

  def save_weights_for_sampler(request, opts) do

    Tinkex.API.post(

      "/api/v1/save_weights_for_sampler",

      request,

      Keyword.put(opts, :pool_type, :training)

    )

  end

end