Skip to content

Commit c226289

Browse files
feat(memreader): update prefermem config (#1294)
* feat: add backend for openai * feat: change memreader for scheaduer * feat: change singlecube llm --------- Co-authored-by: harvey_xiang <harvey_xiang@163.com>
1 parent 93b1014 commit c226289

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

src/memos/mem_reader/multi_modal_struct.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1052,7 +1052,7 @@ def _process_multi_modal_data(
10521052
process_preference_fine,
10531053
non_file_url_fast_items,
10541054
info,
1055-
self.llm,
1055+
self.general_llm,
10561056
self.embedder,
10571057
**kwargs,
10581058
)
@@ -1131,7 +1131,12 @@ def _process_transfer_multi_modal_data(
11311131
)
11321132
# Add preference memory extraction
11331133
future_pref = executor.submit(
1134-
process_preference_fine, non_file_url_nodes, info, self.llm, self.embedder, **kwargs
1134+
process_preference_fine,
1135+
non_file_url_nodes,
1136+
info,
1137+
self.general_llm,
1138+
self.embedder,
1139+
**kwargs,
11351140
)
11361141

11371142
# Collect results

0 commit comments

Comments
 (0)