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"]
|