LennardZuendorf commited on
Commit
d19acd3
·
unverified ·
1 Parent(s): 727303f

fix: fixed token handover bug

Browse files
Files changed (1) hide show
  1. chatmodel.py +1 -0
chatmodel.py CHANGED
@@ -13,6 +13,7 @@ llama_pipeline = pipeline(
13
  model=model,
14
  torch_dtype=torch.float32,
15
  device_map="auto",
 
16
  )
17
 
18
  # Formatting function for message and history
 
13
  model=model,
14
  torch_dtype=torch.float32,
15
  device_map="auto",
16
+ token = token
17
  )
18
 
19
  # Formatting function for message and history