Update modules/pmbl.py
Browse files- 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 = ""
|