Module: Toy::Device

Defined in:
lib/toy/compute.rb,
lib/toy/compute_cuda.rb,
lib/toy/compute_metal.rb

Class Method Summary collapse

Class Method Details

.from_scratch_recipe ⇒ Object 



119
120
121
 
# File 'lib/toy/compute.rb', line 119

def self.from_scratch_recipe
  Toy::LLM::Recipes::FromScratch.new
end

.gpt2_engine ⇒ Object 



115
116
117
 
# File 'lib/toy/compute.rb', line 115

def self.gpt2_engine
  Toy::LLM::Engine::GPT2SeqEngine.new
end

.llama_engine ⇒ Object 



111
112
113
 
# File 'lib/toy/compute.rb', line 111

def self.llama_engine
  Toy::LLM::Engine::LlamaSeqEngine.new
end

.name ⇒ Object 



107
108
109
 
# File 'lib/toy/compute.rb', line 107

def self.name
  "cpu"
end

.warm_start_recipe ⇒ Object 



123
124
125
 
# File 'lib/toy/compute.rb', line 123

def self.warm_start_recipe
  Toy::LLM::Recipes::WarmStart.new
end