Update modeling_aquila.py
Browse files- modeling_aquila.py +1 -1
modeling_aquila.py
CHANGED
|
@@ -931,7 +931,7 @@ class AquilaForCausalLM(AquilaPreTrainedModel):
|
|
| 931 |
topk = 1
|
| 932 |
temperature = 1.0
|
| 933 |
if sft:
|
| 934 |
-
tokens = covert_prompt_to_input_ids_with_history(text, history=[], tokenizer=tokenizer, max_token=
|
| 935 |
tokens = torch.tensor(tokens)[None,].to(device)
|
| 936 |
else :
|
| 937 |
tokens = tokenizer.encode_plus(text)["input_ids"]
|
|
|
|
| 931 |
topk = 1
|
| 932 |
temperature = 1.0
|
| 933 |
if sft:
|
| 934 |
+
tokens = covert_prompt_to_input_ids_with_history(text, history=[], tokenizer=tokenizer, max_token=20480, convo_template=convo_template)
|
| 935 |
tokens = torch.tensor(tokens)[None,].to(device)
|
| 936 |
else :
|
| 937 |
tokens = tokenizer.encode_plus(text)["input_ids"]
|