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
112
112
class ExecuTorchLlmJni : public facebook ::jni::HybridClass<ExecuTorchLlmJni> {
113
113
private:
114
114
friend HybridBase;
115
- float temperature_;
115
+ float temperature_ = 0 . 0f ;
116
116
int model_type_category_;
117
117
std::unique_ptr<llm::IRunner> runner_;
118
118
std::unique_ptr<llm::MultimodalRunner> multi_modal_runner_;
@@ -146,6 +146,7 @@ class ExecuTorchLlmJni : public facebook::jni::HybridClass<ExecuTorchLlmJni> {
146
146
facebook::jni::alias_ref<jstring> tokenizer_path,
147
147
jfloat temperature,
148
148
facebook::jni::alias_ref<jstring> data_path = nullptr ) {
149
+ temperature_ = temperature;
149
150
#if defined(ET_USE_THREADPOOL)
150
151
// Reserve 1 thread for the main thread.
151
152
int32_t num_performant_cores =
You can’t perform that action at this time.
0 commit comments