Code

Sparknlp 876: Introducing LLAMA2 #14148

Code

Merged

maziyarpanahi merged 11 commits intorelease/530-release-candidateJohnSnowLabs/spark-nlp:release/530-release-candidatefrom SPARKNLP-876-Import-and-implement-Llama-2-LlamaForCausalLM-in-ONNXJohnSnowLabs/spark-nlp:SPARKNLP-876-Import-and-implement-Llama-2-LlamaForCausalLM-in-ONNXCopy head branch name to clipboardFeb 6, 2024

+1,331-45

Commits

Commits on Jan 23, 2024

• introducing LLAMA2

  prabod

  committed
  8045e3f

• Added option to read model from model path to onnx wrapper

  prabod

  committed
  492e6c3

• Added option to read model from model path to onnx wrapper

  prabod

  committed
  48cb061

• updated text description

  prabod

  committed
  b4cf4cf

• LLAMA2 python API

  prabod

  committed
  44bbc87

• added method to save onnx_data

  prabod

  committed
  95d2587

• added position ids

  prabod

  committed
  a9b2b5c

• - updated Generate.scala to accept onnx tensors

  Show description for 5304c1f
  prabod

  committed
  5304c1f

• updated max input length

  prabod

  committed
  46e8f15

• updated python default params

  Show description for c0b2c4f
  prabod

  committed
  c0b2c4f

Commits on Jan 31, 2024

• fixed serialization bug

  prabod

  committed
  4dbd0d4