mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-09-24 20:20:37 +00:00
20 lines
535 B
C++
20 lines
535 B
C++
#ifndef __SD_TOKENIZERS_QWEN2_TOKENIZER_H__
|
|
#define __SD_TOKENIZERS_QWEN2_TOKENIZER_H__
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "bpe_tokenizer.h"
|
|
|
|
class Qwen2Tokenizer : public BPETokenizer {
|
|
protected:
|
|
void load_from_merges(const std::string& merges_utf8_str);
|
|
Qwen2Tokenizer(const std::string& merges_utf8_str,
|
|
const std::vector<std::string>& special_tokens_override);
|
|
|
|
public:
|
|
explicit Qwen2Tokenizer(const std::string& merges_utf8_str = "");
|
|
};
|
|
|
|
#endif // __SD_TOKENIZERS_QWEN2_TOKENIZER_H__
|