Lakshay121 commited on
Commit
6f67f6d
·
verified ·
1 Parent(s): 812b647

made changes in max_cache_length

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ demo = gr.Interface(
50
  # Patch Phi-3.5-mini-instruct's `prepare_inputs_for_generation`
51
  def patched_prepare_inputs_for_generation(self, input_ids, past_key_values=None, **kwargs):
52
  if past_key_values is not None:
53
- max_cache_length = past_key_values.get_seq_length() # Fix here
54
  else:
55
  max_cache_length = None
56
  return self._default_prepare_inputs_for_generation(input_ids, past_key_values, **kwargs)
 
50
  # Patch Phi-3.5-mini-instruct's `prepare_inputs_for_generation`
51
  def patched_prepare_inputs_for_generation(self, input_ids, past_key_values=None, **kwargs):
52
  if past_key_values is not None:
53
+ max_cache_length = len(past_key_values[0][0])
54
  else:
55
  max_cache_length = None
56
  return self._default_prepare_inputs_for_generation(input_ids, past_key_values, **kwargs)