Update main.py
Browse files
main.py
CHANGED
@@ -6,7 +6,7 @@ from ctransformers import AutoModelForCausalLM
|
|
6 |
#Model loading
|
7 |
llm = AutoModelForCausalLM.from_pretrained("zephyr-7b-beta.Q4_K_S.gguf",
|
8 |
model_type='mistral',
|
9 |
-
max_new_tokens =
|
10 |
threads = 3,
|
11 |
)
|
12 |
|
|
|
6 |
#Model loading
|
7 |
llm = AutoModelForCausalLM.from_pretrained("zephyr-7b-beta.Q4_K_S.gguf",
|
8 |
model_type='mistral',
|
9 |
+
max_new_tokens = 256,
|
10 |
threads = 3,
|
11 |
)
|
12 |
|