feat: implement the complete bpe function (#119)

* implement the complete bpe function
---------

Co-authored-by: leejet <leejet714@gmail.com>
This commit is contained in:
旺旺碎冰冰andleejet authored and GitHub committed 2023-12-23 12:11:07 +08:00
1 parent 8f6b4a39d6
commit 0e64238e4c
4 files changed
+95 -11

No files matched your search

+4
View File
@@ -5,6 +5,8 @@
#include <string>
#include <vector>
#include "ggml/ggml.h"
enum RNGType {
STD_DEFAULT_RNG,
CUDA_RNG
@@ -42,10 +44,12 @@ public:
bool free_params_immediately = false,
std::string lora_model_dir = "",
RNGType rng_type = STD_DEFAULT_RNG);
bool load_from_file(const std::string& model_path,
const std::string& vae_path,
ggml_type wtype,
Schedule d = DEFAULT);
std::vector<uint8_t*> txt2img(
std::string prompt,
std::string negative_prompt,