Swift · Quickstart

Quickstart

4.0.0-rc.1 · 3ee5e534

import UForm

let executor = try Executor.forRam()
let arena = try Arena(on: executor)                                  // where the outputs land
let model = try Model.loadRam(path: "model.safetensors")
let encoded = try model.on(executor).textEncode(["A blue house"], outputs: .embeddings, arena: arena)
let embedding = Array(encoded.embeddings![0])                        // [Float], the model's dimensions wide