BAAI
/

shunxing1234 commited on
Commit
72995c9
1 Parent(s): dfa1e25

Update modeling_aquila.py

Browse files
Files changed (1) hide show
  1. modeling_aquila.py +2 -1
modeling_aquila.py CHANGED
@@ -507,7 +507,8 @@ class AquilaModel(AquilaPreTrainedModel):
507
  use_cache = use_cache if use_cache is not None else self.config.use_cache
508
 
509
  return_dict = return_dict if return_dict is not None else self.config.use_return_dict
510
-
 
511
  # retrieve input_ids and inputs_embeds
512
  if input_ids is not None and inputs_embeds is not None:
513
  raise ValueError("You cannot specify both decoder_input_ids and decoder_inputs_embeds at the same time")
 
507
  use_cache = use_cache if use_cache is not None else self.config.use_cache
508
 
509
  return_dict = return_dict if return_dict is not None else self.config.use_return_dict
510
+ import pdb
511
+ pdb.set_trace()
512
  # retrieve input_ids and inputs_embeds
513
  if input_ids is not None and inputs_embeds is not None:
514
  raise ValueError("You cannot specify both decoder_input_ids and decoder_inputs_embeds at the same time")