Spaces:
Runtime error
Runtime error
made changes in max_cache_length
Browse files
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
|
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)
|