Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -19,8 +19,8 @@ class EndpointHandler:
|
|
19 |
)
|
20 |
|
21 |
generation_config = model.generation_config
|
22 |
-
generation_config.max_new_tokens =
|
23 |
-
generation_config.temperature = 0
|
24 |
generation_config.num_return_sequences = 1
|
25 |
generation_config.pad_token_id = tokenizer.eos_token_id
|
26 |
generation_config.eos_token_id = tokenizer.eos_token_id
|
|
|
19 |
)
|
20 |
|
21 |
generation_config = model.generation_config
|
22 |
+
generation_config.max_new_tokens = 200
|
23 |
+
generation_config.temperature = 0.7
|
24 |
generation_config.num_return_sequences = 1
|
25 |
generation_config.pad_token_id = tokenizer.eos_token_id
|
26 |
generation_config.eos_token_id = tokenizer.eos_token_id
|