diff --git a/examples/common/common.hpp b/examples/common/common.hpp index c9e8eee..82328bc 100644 --- a/examples/common/common.hpp +++ b/examples/common/common.hpp @@ -809,7 +809,7 @@ struct SDContextParams { } void build_embedding_map() { - static const std::vector valid_ext = {".pt", ".safetensors", ".gguf"}; + static const std::vector valid_ext = {".gguf", ".safetensors", ".pt"}; if (!fs::exists(embedding_dir) || !fs::is_directory(embedding_dir)) { return; @@ -1606,7 +1606,7 @@ struct SDGenerationParams { return; } static const std::regex re(R"(]+):([^>]+)>)"); - static const std::vector valid_ext = {".pt", ".safetensors", ".gguf"}; + static const std::vector valid_ext = {".gguf", ".safetensors", ".pt"}; std::smatch m; std::string tmp = prompt;