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

Show error message when -f fails

Slaren 2 лет назад
Родитель
Сommit
0d054e292e
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      examples/common.cpp

+ 5 - 0
examples/common.cpp

@@ -66,6 +66,11 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
                 break;
             }
             std::ifstream file(argv[i]);
+            if (!file) {
+                fprintf(stderr, "error: failed to open file '%s'\n", argv[i]);
+                invalid_param = true;
+                break;
+            }
             std::copy(std::istreambuf_iterator<char>(file), std::istreambuf_iterator<char>(), back_inserter(params.prompt));
             if (params.prompt.back() == '\n') {
                 params.prompt.pop_back();