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)
|