bindings.cc 219 Bytes
Newer Older
1
#include "models/llama.hpp"
2
#include <pybind11/pybind11.h>
3
4
5

namespace py = pybind11;

6
PYBIND11_MODULE(_infinilm, m) {
7
8
9
10
    m.doc() = "InfiniLM Llama model Python bindings";

    infinilm::models::llama::bind_llama(m);
}