Method: Safetensors::Torch.load_file

Defined in:
lib/safetensors/torch.rb

.load_file(filename, device: "cpu") ⇒ Object 



30
31
32
33
34
35
36
37
38
 
# File 'lib/safetensors/torch.rb', line 30

def load_file(filename, device: "cpu")
  result = {}
  Safetensors.safe_open(filename, framework: "torch", device: device) do |f|
    f.keys.each do |k|
      result[k] = f.get_tensor(k)
    end
  end
  result
end