admin/llgo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

runtime: Float, Time

Browse Source
This commit is contained in:
xushiwei
2024-05-07 07:32:25 +08:00
parent 025cff9494
commit 0edeb5cfd0
3 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
_demo/llama2-c/run.go
View File

@@ -25,9 +25,9 @@ func main() {
var prompt *c.Char = c.Str("Once upon a time")
var checkpointPath *c.Char = c.Str("stories15M.bin")
var tokenizerPath *c.Char = c.Str("tokenizer.bin")
var temperature, topp llama2.Float = 1.0, 0.9
var temperature, topp c.Float = 1.0, 0.9
var steps c.Int = 256
var rngSeed uint64 = uint64(llama2.Time(nil))
var rngSeed uint64 = uint64(c.Time(nil))
// build the Transformer via the model .bin file
var transformer llama2.Transformer