Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -396,7 +396,8 @@ class Phi3MiniEducationalLLM(Runnable):
|
|
| 396 |
repetition_penalty=1.1,
|
| 397 |
pad_token_id=self.tokenizer.eos_token_id,
|
| 398 |
early_stopping=True,
|
| 399 |
-
use_cache=
|
|
|
|
| 400 |
)
|
| 401 |
|
| 402 |
# Decode only new tokens
|
|
@@ -467,7 +468,8 @@ class Phi3MiniEducationalLLM(Runnable):
|
|
| 467 |
"repetition_penalty": 1.2,
|
| 468 |
"pad_token_id": self.tokenizer.eos_token_id,
|
| 469 |
"streamer": streamer,
|
| 470 |
-
"use_cache":
|
|
|
|
| 471 |
}
|
| 472 |
|
| 473 |
# Start generation in background
|
|
|
|
| 396 |
repetition_penalty=1.1,
|
| 397 |
pad_token_id=self.tokenizer.eos_token_id,
|
| 398 |
early_stopping=True,
|
| 399 |
+
use_cache=False
|
| 400 |
+
past_key_values=None
|
| 401 |
)
|
| 402 |
|
| 403 |
# Decode only new tokens
|
|
|
|
| 468 |
"repetition_penalty": 1.2,
|
| 469 |
"pad_token_id": self.tokenizer.eos_token_id,
|
| 470 |
"streamer": streamer,
|
| 471 |
+
"use_cache": False
|
| 472 |
+
"past_key_values": None
|
| 473 |
}
|
| 474 |
|
| 475 |
# Start generation in background
|