File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -112,7 +112,7 @@ class ExecuTorchLlmCallbackJni
112112class ExecuTorchLlmJni : public facebook ::jni::HybridClass<ExecuTorchLlmJni> {
113113 private:
114114 friend HybridBase;
115- float temperature_;
115+ float temperature_ = 0 . 0f ;
116116 int model_type_category_;
117117 std::unique_ptr<llm::IRunner> runner_;
118118 std::unique_ptr<llm::MultimodalRunner> multi_modal_runner_;
@@ -146,6 +146,7 @@ class ExecuTorchLlmJni : public facebook::jni::HybridClass<ExecuTorchLlmJni> {
146146 facebook::jni::alias_ref<jstring> tokenizer_path,
147147 jfloat temperature,
148148 facebook::jni::alias_ref<jstring> data_path = nullptr ) {
149+ temperature_ = temperature;
149150#if defined(ET_USE_THREADPOOL)
150151 // Reserve 1 thread for the main thread.
151152 int32_t num_performant_cores =
You can’t perform that action at this time.
0 commit comments