Commit bbf3e3a
Changed files (2)
ext
elelem
llama2
lib
elelem
llama2
ext/elelem/llama2/extconf.rb
@@ -11,8 +11,8 @@ NATIVE = File.join(GEM_ROOT, "ext", "elelem", "llama2", "native")
class BuildError < StandardError; end
def run(*cmd)
- warn "elelem-llama2: + #{cmd.join(' ')}"
- system(*cmd) || raise(BuildError, "build step failed: #{cmd.join(' ')}")
+ warn "elelem-llama2: + #{cmd.join(" ")}"
+ system(*cmd) || raise(BuildError, "build step failed: #{cmd.join(" ")}")
end
unless File.exist?(File.join(VENDOR, "run.c"))
lib/elelem/llama2/provider.rb
@@ -25,7 +25,7 @@ module Elelem
# vendor/llama2.c/export.py + tokenizer.bin), not GGUF.
def initialize(checkpoint:, tokenizer:, steps: 0, temperature: 1.0, topp: 0.9, seed: -1)
@steps = steps
- seed = seed.negative? ? Process.clock_gettime(Process::CLOCK_REALTIME, :nanosecond) : seed
+ seed = Process.clock_gettime(Process::CLOCK_REALTIME, :nanosecond) if seed.negative?
@handle = self.class.functions[:open].call(checkpoint, tokenizer, temperature, topp, seed)
raise "llama2: failed to load checkpoint at #{checkpoint}" if @handle.null?