Просмотр исходного кода

quantize : fail fast on write errors (#3521)

cebtenzzre 2 лет назад
Родитель
Сommit
f1782c68de
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      llama.cpp

+ 1 - 0
llama.cpp

@@ -7194,6 +7194,7 @@ static void llama_model_quantize_internal(const std::string & fname_inp, const s
     }
     }
 
 
     std::ofstream fout(fname_out, std::ios::binary);
     std::ofstream fout(fname_out, std::ios::binary);
+    fout.exceptions(std::ofstream::failbit); // fail fast on write errors
 
 
     const size_t meta_size = gguf_get_meta_size(ctx_out);
     const size_t meta_size = gguf_get_meta_size(ctx_out);