File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change 1111#include < cstring>
1212#include < functional>
1313#include < map>
14+ #include < regex>
1415#include < sstream>
1516#include < stdexcept>
1617
@@ -1464,7 +1465,8 @@ bool llama_model::load_tensors(llama_model_loader & ml) {
14641465 if (ml.tensor_buft_overrides ) {
14651466 std::string tensor_name = tn.str ();
14661467 for (const auto * overrides = ml.tensor_buft_overrides ; overrides->pattern != nullptr ; ++overrides) {
1467- if (tensor_name.find (overrides->pattern ) != std::string::npos) {
1468+ std::regex pattern (overrides->pattern );
1469+ if (std::regex_search (tensor_name, pattern)) {
14681470 LLAMA_LOG_DEBUG (" tensor %s buffer type overriden to %s\n " , tensor_name.c_str (), ggml_backend_buft_name (overrides->buft ));
14691471 buft = overrides->buft ;
14701472 break ;
You can’t perform that action at this time.
0 commit comments