Sergidev commited on
Commit
626324b
1 Parent(s): 417b8dc

Update modules/pmbl.py

Browse files
Files changed (1) hide show
  1. modules/pmbl.py +2 -0
modules/pmbl.py CHANGED
@@ -111,6 +111,8 @@ class PMBL:
111
  stop=["</s>", "\nUser:", "\nuser:", "\nSystem:", "\nsystem:"],
112
  echo=False,
113
  stream=True
 
 
114
  )
115
 
116
  response_text = ""
 
111
  stop=["</s>", "\nUser:", "\nuser:", "\nSystem:", "\nsystem:"],
112
  echo=False,
113
  stream=True
114
+ use_flash_attn=True
115
+ mlock=True,
116
  )
117
 
118
  response_text = ""