Skip to content

Commit

Permalink
Removed testing definitions in sample
Browse files Browse the repository at this point in the history
  • Loading branch information
AsyaPronina committed Nov 30, 2024
1 parent b2fc44b commit 7c8ff06
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
2 changes: 1 addition & 1 deletion samples/cpp/chat_sample/chat_sample.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ int main(int argc, char* argv[]) try {
std::string prompt;
std::string models_path = argv[1];

std::string device = "NPU"; // GPU, NPU can be used as well
std::string device = "CPU"; // GPU, NPU can be used as well
ov::genai::LLMPipeline pipe(models_path, device);

ov::genai::GenerationConfig config;
Expand Down
4 changes: 4 additions & 0 deletions src/cpp/src/llm_pipeline_static.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -644,6 +644,10 @@ SMStaticLLMPipeline::SMStaticLLMPipeline(
update_config(properties, {"NPUW_LLM_MAX_PROMPT_LEN", kMaxPromptLen});
update_config(properties, {"NPUW_LLM_MIN_RESPONSE_LEN", kMinResponseLen});
update_config(properties, {"NPUW_LLM_GENERATE_HINT", generate_hint});

// FIXME: Support CACHE_DIR in future
drop_cache_dir(properties);

auto compiled = core.compile_model(model, "NPU", properties);

m_request = compiled.create_infer_request();
Expand Down

0 comments on commit 7c8ff06

Please sign in to comment.