BAAI
/

ldwang commited on
Commit
7cd14e1
·
1 Parent(s): 6f06033

Update modeling_aquila.py

Browse files
Files changed (1) hide show
  1. 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=2048, convo_template=convo_template)
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"]