Archives/gpt4all
2
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all synced 2024-11-08 07:10:32 +00:00
Code Issues Packages Projects Releases Wiki Activity
825fa64c17
gpt4all/gpt4all-backend/llamamodel_impl.h

39 lines
1.3 KiB
C
Raw Normal View History

Dlopen backend 5 (#779) Major change to the backend that allows for pluggable versions of llama.cpp/ggml. This was squashed merged from dlopen_backend_5 where the history is preserved.
2023-05-31 21:04:01 +00:00
#ifndef LLAMAMODEL_H_I_KNOW_WHAT_I_AM_DOING_WHEN_INCLUDING_THIS_FILE
#error This file is NOT meant to be included outside of llamamodel.cpp. Doing so is DANGEROUS. Be sure to know what you are doing before proceeding to #define LLAMAMODEL_H_I_KNOW_WHAT_I_AM_DOING_WHEN_INCLUDING_THIS_FILE
#endif
Add llama.cpp support for loading llama based models in the gui. We now support loading both gptj derived models and llama derived models.
2023-04-15 19:57:32 +00:00
#ifndef LLAMAMODEL_H
#define LLAMAMODEL_H
#include <string>
#include <functional>
#include <vector>
#include "llmodel.h"
Dlopen backend 5 (#779) Major change to the backend that allows for pluggable versions of llama.cpp/ggml. This was squashed merged from dlopen_backend_5 where the history is preserved.
2023-05-31 21:04:01 +00:00
struct LLamaPrivate;
Add llama.cpp support for loading llama based models in the gui. We now support loading both gptj derived models and llama derived models.
2023-04-15 19:57:32 +00:00
class LLamaModel : public LLModel {
public:
LLamaModel();
~LLamaModel();
bool loadModel(const std::string &modelPath) override;
bool isModelLoaded() const override;
First attempt at providing a persistent chat list experience. Limitations: 1) Context is not restored for gpt-j models 2) When you switch between different model types in an existing chat the context and all the conversation is lost 3) The settings are not chat or conversation specific 4) The sizes of the chat persisted files are very large due to how much data the llama.cpp backend tries to persist. Need to investigate how we can shrink this.
2023-05-04 19:31:41 +00:00
size_t stateSize() const override;
size_t saveState(uint8_t *dest) const override;
size_t restoreState(const uint8_t *src) override;
Add llama.cpp support for loading llama based models in the gui. We now support loading both gptj derived models and llama derived models.
2023-04-15 19:57:32 +00:00
void setThreadCount(int32_t n_threads) override;
llmodel: constify LLModel::threadCount()
2023-05-21 20:45:29 +00:00
int32_t threadCount() const override;
Add llama.cpp support for loading llama based models in the gui. We now support loading both gptj derived models and llama derived models.
2023-04-15 19:57:32 +00:00
private:
LLamaPrivate *d_ptr;
Backend prompt dedup (#822) * Deduplicated prompt() function code
2023-06-04 12:59:24 +00:00
protected:
std::vector<Token> tokenize(const std::string&) const override;
std::string_view tokenToString(Token) const override;
Token sampleToken(PromptContext& ctx) const override;
bool evalTokens(PromptContext& ctx, const std::vector<int32_t> &tokens) const override;
int32_t contextLength() const override;
const std::vector<Token>& endTokens() const override;
Add llama.cpp support for loading llama based models in the gui. We now support loading both gptj derived models and llama derived models.
2023-04-15 19:57:32 +00:00
};
Dlopen backend 5 (#779) Major change to the backend that allows for pluggable versions of llama.cpp/ggml. This was squashed merged from dlopen_backend_5 where the history is preserved.
2023-05-31 21:04:01 +00:00
#endif // LLAMAMODEL_H
Reference in New Issue Copy Permalink