Update modeling_InternLM_XComposer.py
Browse files
modeling_InternLM_XComposer.py
CHANGED
|
@@ -36,12 +36,12 @@ class InternLMXComposerForCausalLM(PreTrainedModel):
|
|
| 36 |
|
| 37 |
gen_config = dict(
|
| 38 |
num_beams=5,
|
| 39 |
-
do_sample=
|
| 40 |
min_length=1,
|
| 41 |
repetition_penalty=1.5,
|
| 42 |
length_penalty=1.0,
|
| 43 |
temperature=0.2,
|
| 44 |
-
max_new_tokens=
|
| 45 |
)
|
| 46 |
|
| 47 |
def __init__(self, config):
|
|
|
|
| 36 |
|
| 37 |
gen_config = dict(
|
| 38 |
num_beams=5,
|
| 39 |
+
do_sample=True,
|
| 40 |
min_length=1,
|
| 41 |
repetition_penalty=1.5,
|
| 42 |
length_penalty=1.0,
|
| 43 |
temperature=0.2,
|
| 44 |
+
max_new_tokens=1000,
|
| 45 |
)
|
| 46 |
|
| 47 |
def __init__(self, config):
|