Spaces:
Sleeping
Sleeping
Commit ·
6c497f2
1
Parent(s): 19767b2
setted the n_threads
Browse files
main.py
CHANGED
|
@@ -6,6 +6,7 @@ import logging
|
|
| 6 |
llm = Llama(
|
| 7 |
model_path="./quantized_model/sql_gpt_quantized.gguf",
|
| 8 |
n_ctx=512,
|
|
|
|
| 9 |
n_gpu_layers=-1
|
| 10 |
)
|
| 11 |
|
|
|
|
| 6 |
llm = Llama(
|
| 7 |
model_path="./quantized_model/sql_gpt_quantized.gguf",
|
| 8 |
n_ctx=512,
|
| 9 |
+
n_threads=1,
|
| 10 |
n_gpu_layers=-1
|
| 11 |
)
|
| 12 |
|