Ley_Fill7 commited on
Commit
78c193e
·
1 Parent(s): a9c90e6

Added callback manager handler

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -33,7 +33,10 @@ embed_model = NVIDIAEmbeddings(
33
  def custom_resolve_embed_model(embed_model):
34
  if isinstance(embed_model, NVIDIAEmbeddings):
35
  return embed_model
36
- return resolve_embed_model(embed_model)
 
 
 
37
 
38
  reranker = NVIDIARerank(
39
  model="nvidia/nv-rerankqa-mistral-4b-v3",
 
33
  def custom_resolve_embed_model(embed_model):
34
  if isinstance(embed_model, NVIDIAEmbeddings):
35
  return embed_model
36
+ embed_model = resolve_embed_model(embed_model)
37
+ if not isinstance(embed_model, NVIDIAEmbeddings):
38
+ embed_model.callback_manager = Settings.callback_manager
39
+ return embed_model
40
 
41
  reranker = NVIDIARerank(
42
  model="nvidia/nv-rerankqa-mistral-4b-v3",